ONE - On-device Neural Engine
|
#include <Pack.h>
Public Member Functions | |
Pack (std::vector< const Tensor * > inputs, Tensor *output, const PackParams ¶ms) | |
const Tensor * | input (int index) const |
Tensor * | output () const |
void | configure () override |
void | execute () const override |
Public Member Functions inherited from luci_interpreter::KernelWithParams< PackParams > | |
const PackParams & | params () const |
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::KernelWithParams< PackParams > | |
KernelWithParams (std::vector< const Tensor * > inputs, std::vector< Tensor * > outputs, const PackParams ¶ms) | |
Protected Member Functions inherited from luci_interpreter::Kernel | |
Kernel (std::vector< const Tensor * > inputs, std::vector< Tensor * > outputs) | |
Protected Attributes inherited from luci_interpreter::KernelWithParams< PackParams > | |
const PackParams | _params |
Protected Attributes inherited from luci_interpreter::Kernel | |
const std::vector< const Tensor * > | _inputs |
const std::vector< Tensor * > | _outputs |
luci_interpreter::kernels::Pack::Pack | ( | std::vector< const Tensor * > | inputs, |
Tensor * | output, | ||
const PackParams & | params | ||
) |
Definition at line 30 of file Pack.cpp.
References output().
|
overridevirtual |
Implements luci_interpreter::Kernel.
Definition at line 35 of file Pack.cpp.
References luci_interpreter::Kernel::_inputs, luci_interpreter::PackParams::axis, luci_interpreter::Shape::dim(), luci_interpreter::Tensor::element_type(), LUCI_INTERPRETER_CHECK, luci_interpreter::Shape::num_dims(), output(), output_shape, luci_interpreter::KernelWithParams< PackParams >::params(), luci_interpreter::Tensor::resize(), luci_interpreter::Tensor::scale(), luci_interpreter::Tensor::shape(), luci_interpreter::PackParams::values_count, and luci_interpreter::Tensor::zero_point().
|
overridevirtual |
Implements luci_interpreter::Kernel.
Definition at line 96 of file Pack.cpp.
References luci_interpreter::Kernel::_inputs.
|
inline |
Definition at line 33 of file Pack.h.
References luci_interpreter::Kernel::_inputs.
|
inline |
Definition at line 34 of file Pack.h.
References luci_interpreter::Kernel::_outputs.
Referenced by configure(), and Pack().