ONE - On-device Neural Engine
|
#include <AveragePool2D.h>
Public Member Functions | |
AveragePool2D (const Tensor *input, Tensor *output, Tensor *scratchpad, const Pool2DParams ¶ms) | |
const Tensor * | input () const |
Tensor * | output () const |
void | configure () override |
void | execute () const override |
Public Member Functions inherited from luci_interpreter::KernelWithParams< Pool2DParams > | |
const Pool2DParams & | 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< Pool2DParams > | |
KernelWithParams (std::vector< const Tensor * > inputs, std::vector< Tensor * > outputs, const Pool2DParams ¶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< Pool2DParams > | |
const Pool2DParams | _params |
Protected Attributes inherited from luci_interpreter::Kernel | |
const std::vector< const Tensor * > | _inputs |
const std::vector< Tensor * > | _outputs |
Definition at line 28 of file AveragePool2D.h.
luci_interpreter::kernels::AveragePool2D::AveragePool2D | ( | const Tensor * | input, |
Tensor * | output, | ||
Tensor * | scratchpad, | ||
const Pool2DParams & | params | ||
) |
Definition at line 31 of file AveragePool2D.cpp.
References input().
|
overridevirtual |
Implements luci_interpreter::Kernel.
Definition at line 37 of file AveragePool2D.cpp.
References luci_interpreter::KernelWithParams< Pool2DParams >::_params, luci_interpreter::kernels::computeOutputSize(), luci_interpreter::kernels::computePadding(), luci_interpreter::Shape::dim(), luci_interpreter::Pool2DParams::filter_height, luci_interpreter::Pool2DParams::filter_width, luci_interpreter::Kernel::getOutputTensors(), luci_interpreter::kernels::getTensorShape(), input(), LUCI_INTERPRETER_CHECK, output(), luci_interpreter::Pool2DParams::padding, luci_interpreter::Tensor::resize(), luci_interpreter::Tensor::shape(), luci_interpreter::Pool2DParams::stride_height, and luci_interpreter::Pool2DParams::stride_width.
|
overridevirtual |
Implements luci_interpreter::Kernel.
Definition at line 85 of file AveragePool2D.cpp.
References input().
|
inline |
Definition at line 34 of file AveragePool2D.h.
References luci_interpreter::Kernel::_inputs.
Referenced by AveragePool2D(), configure(), and execute().
|
inline |
Definition at line 35 of file AveragePool2D.h.
References luci_interpreter::Kernel::_outputs.
Referenced by configure().