ONE - On-device Neural Engine
|
#include <SplitV.h>
Public Member Functions | |
SplitV (const Tensor *input, const Tensor *size_splits, const Tensor *axis, std::vector< Tensor * > outputs) | |
const Tensor * | input () const |
const Tensor * | size_splits () const |
const Tensor * | axis () const |
Tensor * | output (int index) const |
void | configure () override |
void | execute () const override |
Public Member Functions inherited from luci_interpreter::Kernel | |
virtual | ~Kernel ()=default |
const std::vector< const Tensor * > & | getInputTensors () const |
const std::vector< Tensor * > & | getOutputTensors () const |
Additional Inherited Members | |
Protected Member Functions inherited from luci_interpreter::Kernel | |
Kernel (std::vector< const Tensor * > inputs, std::vector< Tensor * > outputs) | |
Protected Attributes inherited from luci_interpreter::Kernel | |
const std::vector< const Tensor * > | _inputs |
const std::vector< Tensor * > | _outputs |
luci_interpreter::kernels::SplitV::SplitV | ( | const Tensor * | input, |
const Tensor * | size_splits, | ||
const Tensor * | axis, | ||
std::vector< Tensor * > | outputs | ||
) |
Definition at line 28 of file SplitV.cpp.
References axis(), input(), and size_splits().
|
inline |
Definition at line 36 of file SplitV.h.
References luci_interpreter::Kernel::_inputs.
Referenced by configure(), and SplitV().
|
overridevirtual |
Implements luci_interpreter::Kernel.
Definition at line 34 of file SplitV.cpp.
References luci_interpreter::Kernel::_outputs, axis(), luci_interpreter::Shape::dim(), input(), luci_interpreter::Shape::num_dims(), output_shape, luci_interpreter::RuntimeShape::resize(), luci_interpreter::Tensor::shape(), and size_splits().
|
overridevirtual |
Implements luci_interpreter::Kernel.
Definition at line 80 of file SplitV.cpp.
References luci_interpreter::Kernel::_outputs, input(), and TF_LITE_SPLIT.
|
inline |
Definition at line 34 of file SplitV.h.
References luci_interpreter::Kernel::_inputs.
Referenced by configure(), execute(), and SplitV().
|
inline |
|
inline |
Definition at line 35 of file SplitV.h.
References luci_interpreter::Kernel::_inputs.
Referenced by configure(), and SplitV().