ONE - On-device Neural Engine
|
#include <Layer.h>
Public Member Functions | |
Layer (const std::shared_ptr< ExternalContext > external_context) | |
~Layer () | |
void | prepare () override |
virtual bool | create ()=0 |
virtual bool | setup ()=0 |
Public Member Functions inherited from onert::exec::IFunction | |
virtual | ~IFunction ()=default |
virtual void | run ()=0 |
Protected Attributes | |
xnn_operator_t | _kernel_op |
bool | _create |
bool | _setup |
const std::shared_ptr< ExternalContext > | _external_context |
|
inline |
Definition at line 43 of file Layer.h.
|
inline |
Definition at line 49 of file Layer.h.
References _kernel_op.
|
pure virtual |
Implemented in onert::backend::xnnpack::ops::ConvolutionLayer, onert::backend::xnnpack::ops::DepthwiseConvolutionLayer, and onert::backend::xnnpack::ops::FullyConnectedLayer.
Referenced by prepare().
|
inlineoverridevirtual |
Reimplemented from onert::exec::IFunction.
|
pure virtual |
Implemented in onert::backend::xnnpack::ops::ConvolutionLayer, onert::backend::xnnpack::ops::DepthwiseConvolutionLayer, and onert::backend::xnnpack::ops::FullyConnectedLayer.
Referenced by prepare().
|
protected |
|
protected |
Definition at line 73 of file Layer.h.
Referenced by onert::backend::xnnpack::ops::ConvolutionLayer::run(), onert::backend::xnnpack::ops::DepthwiseConvolutionLayer::run(), onert::backend::xnnpack::ops::FullyConnectedLayer::run(), onert::backend::xnnpack::ops::ConvolutionLayer::setup(), onert::backend::xnnpack::ops::DepthwiseConvolutionLayer::setup(), and onert::backend::xnnpack::ops::FullyConnectedLayer::setup().
|
protected |
Definition at line 70 of file Layer.h.
Referenced by onert::backend::xnnpack::ops::ConvolutionLayer::create(), onert::backend::xnnpack::ops::DepthwiseConvolutionLayer::create(), onert::backend::xnnpack::ops::FullyConnectedLayer::create(), onert::backend::xnnpack::ops::ConvolutionLayer::run(), onert::backend::xnnpack::ops::DepthwiseConvolutionLayer::run(), onert::backend::xnnpack::ops::FullyConnectedLayer::run(), onert::backend::xnnpack::ops::ConvolutionLayer::setup(), onert::backend::xnnpack::ops::DepthwiseConvolutionLayer::setup(), onert::backend::xnnpack::ops::FullyConnectedLayer::setup(), and ~Layer().
|
protected |
Definition at line 72 of file Layer.h.
Referenced by prepare(), onert::backend::xnnpack::ops::ConvolutionLayer::run(), onert::backend::xnnpack::ops::DepthwiseConvolutionLayer::run(), and onert::backend::xnnpack::ops::FullyConnectedLayer::run().