ONE - On-device Neural Engine
|
Pass to convert dynamic batch to single batch. More...
#include <DynamicBatchToSingleBatchPass.h>
Public Member Functions | |
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 convert dynamic batch to single batch.
Definition at line 28 of file DynamicBatchToSingleBatchPass.h.
|
inlinevirtual |
Reimplemented from logo::Pass.
Definition at line 31 of file DynamicBatchToSingleBatchPass.h.
|
virtual |
Run the pass.
Implements logo::Pass.
Definition at line 25 of file DynamicBatchToSingleBatchPass.cpp.
References luci::FixedArityNode< N, Base >::at(), loco::TensorShape::dim(), luci::CircleInput::index(), luci::input_node(), loco::input_nodes(), loco::TensorShape::rank(), and loco::Dimension::set().
Referenced by package.infer.session::inference().