ONE - On-device Neural Engine
|
#include <OneHot.h>
Public Member Functions | |
OneHot (const Tensor *indices, const Tensor *depth, const Tensor *on_value, const Tensor *off_value, Tensor *output, const OneHotParams ¶ms) | |
const Tensor * | indices () const |
const Tensor * | depth () const |
const Tensor * | on_value () const |
const Tensor * | off_value () const |
Tensor * | output () const |
void | configure () override |
void | execute () const override |
OneHot (const Tensor *indices, const Tensor *depth, const Tensor *on_value, const Tensor *off_value, Tensor *output, const OneHotParams ¶ms) | |
const Tensor * | indices () const |
const Tensor * | depth () const |
const Tensor * | on_value () const |
const Tensor * | off_value () const |
Tensor * | output () const |
void | configure () override |
void | execute () const override |
Public Member Functions inherited from luci_interpreter::KernelWithParams< OneHotParams > | |
const OneHotParams & | 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< OneHotParams > | |
KernelWithParams (std::vector< const Tensor * > inputs, std::vector< Tensor * > outputs, const OneHotParams ¶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< OneHotParams > | |
const OneHotParams | _params |
Protected Attributes inherited from luci_interpreter::Kernel | |
const std::vector< const Tensor * > | _inputs |
const std::vector< Tensor * > | _outputs |
luci_interpreter::kernels::OneHot::OneHot | ( | const Tensor * | indices, |
const Tensor * | depth, | ||
const Tensor * | on_value, | ||
const Tensor * | off_value, | ||
Tensor * | output, | ||
const OneHotParams & | params | ||
) |
Definition at line 69 of file OneHot.cpp.
References depth(), indices(), off_value(), and on_value().
luci_interpreter::kernels::OneHot::OneHot | ( | const Tensor * | indices, |
const Tensor * | depth, | ||
const Tensor * | on_value, | ||
const Tensor * | off_value, | ||
Tensor * | output, | ||
const OneHotParams & | params | ||
) |
|
overridevirtual |
Implements luci_interpreter::Kernel.
Definition at line 76 of file OneHot.cpp.
References luci_interpreter::OneHotParams::axis, depth(), indices(), LUCI_INTERPRETER_CHECK, luci_interpreter::Shape::num_dims(), off_value(), on_value(), output(), output_shape, luci_interpreter::KernelWithParams< OneHotParams >::params(), luci_interpreter::Tensor::resize(), and luci_interpreter::Tensor::shape().
|
overridevirtual |
Implements luci_interpreter::Kernel.
|
inline |
Definition at line 35 of file OneHot.h.
References luci_interpreter::Kernel::_inputs.
Referenced by configure(), execute(), and OneHot().
|
inline |
|
overridevirtual |
Implements luci_interpreter::Kernel.
Definition at line 112 of file OneHot.cpp.
References luci_interpreter::OneHotParams::axis, depth(), indices(), off_value(), on_value(), output(), and luci_interpreter::KernelWithParams< OneHotParams >::params().
|
overridevirtual |
Implements luci_interpreter::Kernel.
|
inline |
Definition at line 34 of file OneHot.h.
References luci_interpreter::Kernel::_inputs.
Referenced by configure(), execute(), and OneHot().
|
inline |
|
inline |
Definition at line 37 of file OneHot.h.
References luci_interpreter::Kernel::_inputs.
Referenced by configure(), execute(), and OneHot().
|
inline |
|
inline |
Definition at line 36 of file OneHot.h.
References luci_interpreter::Kernel::_inputs.
Referenced by configure(), execute(), and OneHot().
|
inline |
|
inline |
Definition at line 39 of file OneHot.h.
References luci_interpreter::Kernel::_outputs.
Referenced by configure(), and execute().
|
inline |