ONE - On-device Neural Engine
|
Pass to copy quantparam (scale, zerop) of a tensor to another tensor. More...
#include <CopyQuantParamPass.h>
Public Types | |
using | TensorVector = std::vector< std::string > |
Public Member Functions | |
CopyQuantParamPass (TensorVector &src_tensors, TensorVector &dst_tensors) | |
virtual const char * | name (void) const |
bool | run (loco::Graph *graph) |
Run the pass. | |
Public Member Functions inherited from logo::Pass | |
virtual | ~Pass ()=default |
Pass to copy quantparam (scale, zerop) of a tensor to another tensor.
Definition at line 30 of file CopyQuantParamPass.h.
using luci::CopyQuantParamPass::TensorVector = std::vector<std::string> |
Definition at line 33 of file CopyQuantParamPass.h.
|
inline |
Definition at line 36 of file CopyQuantParamPass.h.
|
inlinevirtual |
Reimplemented from logo::Pass.
Definition at line 41 of file CopyQuantParamPass.h.
Referenced by run().
|
virtual |
Run the pass.
Implements logo::Pass.
Definition at line 35 of file CopyQuantParamPass.cpp.
References loco::active_nodes(), luci::copy_quantparam(), INFO, LOGGER, name(), and loco::output_nodes().
Referenced by package.infer.session::inference().