ONE - On-device Neural Engine
|
#include <GRU.h>
Public Member Functions | |
GRU (const Tensor *input, const Tensor *hidden_hidden, const Tensor *hidden_hidden_bias, const Tensor *hidden_input, const Tensor *hidden_input_bias, const Tensor *state, Tensor *output, const GRUParams ¶ms) | |
const Tensor * | input () const |
const Tensor * | hidden_hidden () const |
const Tensor * | hidden_hidden_bias () const |
const Tensor * | hidden_input () const |
const Tensor * | hidden_input_bias () const |
const Tensor * | state () const |
Tensor * | output () const |
void | configure () override |
void | execute () const override |
Public Member Functions inherited from luci_interpreter::KernelWithParams< GRUParams > | |
const GRUParams & | 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< GRUParams > | |
KernelWithParams (std::vector< const Tensor * > inputs, std::vector< Tensor * > outputs, const GRUParams ¶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< GRUParams > | |
const GRUParams | _params |
Protected Attributes inherited from luci_interpreter::Kernel | |
const std::vector< const Tensor * > | _inputs |
const std::vector< Tensor * > | _outputs |
luci_interpreter::kernels::GRU::GRU | ( | const Tensor * | input, |
const Tensor * | hidden_hidden, | ||
const Tensor * | hidden_hidden_bias, | ||
const Tensor * | hidden_input, | ||
const Tensor * | hidden_input_bias, | ||
const Tensor * | state, | ||
Tensor * | output, | ||
const GRUParams & | params | ||
) |
Definition at line 28 of file GRU.cpp.
References hidden_hidden(), hidden_hidden_bias(), hidden_input(), hidden_input_bias(), input(), and state().
|
overridevirtual |
Implements luci_interpreter::Kernel.
Definition at line 37 of file GRU.cpp.
References luci_interpreter::kernels::getTensorShape(), hidden_hidden(), hidden_input(), input(), LUCI_INTERPRETER_CHECK, output(), luci_interpreter::Tensor::resize(), and state().
|
overridevirtual |
Implements luci_interpreter::Kernel.
Definition at line 48 of file GRU.cpp.
References input().
|
inline |
Definition at line 36 of file GRU.h.
References luci_interpreter::Kernel::_inputs.
Referenced by configure(), and GRU().
|
inline |
Definition at line 37 of file GRU.h.
References luci_interpreter::Kernel::_inputs.
Referenced by GRU().
|
inline |
Definition at line 38 of file GRU.h.
References luci_interpreter::Kernel::_inputs.
Referenced by configure(), and GRU().
|
inline |
Definition at line 39 of file GRU.h.
References luci_interpreter::Kernel::_inputs.
Referenced by GRU().
|
inline |
Definition at line 35 of file GRU.h.
References luci_interpreter::Kernel::_inputs.
Referenced by configure(), execute(), and GRU().
|
inline |
Definition at line 41 of file GRU.h.
References luci_interpreter::Kernel::_outputs.
Referenced by configure().
|
inline |
Definition at line 40 of file GRU.h.
References luci_interpreter::Kernel::_inputs.
Referenced by configure(), and GRU().