ONE - On-device Neural Engine
|
#include <PRelu.h>
Public Member Functions | |
PRelu (const Tensor *input, const Tensor *alpha, Tensor *output) | |
~PRelu () | |
const Tensor * | input () const |
const Tensor * | alpha () const |
Tensor * | output () 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::PRelu::PRelu | ( | const Tensor * | input, |
const Tensor * | alpha, | ||
Tensor * | output | ||
) |
Definition at line 33 of file PRelu.cpp.
luci_interpreter::kernels::PRelu::~PRelu | ( | ) |
|
inline |
Definition at line 38 of file PRelu.h.
References luci_interpreter::Kernel::_inputs.
Referenced by configure(), and PRelu().
|
overridevirtual |
Implements luci_interpreter::Kernel.
Definition at line 43 of file PRelu.cpp.
References alpha(), luci_interpreter::kernels::calculateShapeForBroadcast(), luci_interpreter::kernels::getQuantizedConvolutionMultiplers(), input(), LUCI_INTERPRETER_CHECK, luci_interpreter::Shape::num_dims(), output(), luci_interpreter::kernels::quantizeMultiplier(), luci_interpreter::kernels::quantizeMultipliers(), luci_interpreter::Tensor::resize(), luci_interpreter::Tensor::scale(), luci_interpreter::Tensor::shape(), size, and luci_interpreter::Tensor::zero_points().
|
overridevirtual |
Implements luci_interpreter::Kernel.
Definition at line 92 of file PRelu.cpp.
References input().
|
inline |
Definition at line 37 of file PRelu.h.
References luci_interpreter::Kernel::_inputs.
Referenced by configure(), execute(), and PRelu().
|
inline |
Definition at line 39 of file PRelu.h.
References luci_interpreter::Kernel::_outputs.
Referenced by configure().