ONE - On-device Neural Engine
Loading...
Searching...
No Matches
luci_interpreter::kernels::DownsamplingConv2DKernel Class Reference

#include <ConvolutionCommon.h>

Public Member Functions

 DownsamplingConv2DKernel ()=delete
 
 DownsamplingConv2DKernel (const circle::Operator *cur_op, BaseRuntimeGraph *runtime_graph)
 
const circle::Tensor * input () const
 
const circle::Tensor * filter () const
 
const circle::Tensor * bias () const
 
const circle::Tensor * output () const
 
BaseRuntimeGraphruntime_graph () const
 

Detailed Description

Definition at line 38 of file ConvolutionCommon.h.

Constructor & Destructor Documentation

◆ DownsamplingConv2DKernel() [1/2]

luci_interpreter::kernels::DownsamplingConv2DKernel::DownsamplingConv2DKernel ( )
delete

◆ DownsamplingConv2DKernel() [2/2]

luci_interpreter::kernels::DownsamplingConv2DKernel::DownsamplingConv2DKernel ( const circle::Operator *  cur_op,
BaseRuntimeGraph runtime_graph 
)
inlineexplicit

Definition at line 43 of file ConvolutionCommon.h.

44 : _runtime_graph(runtime_graph)
45 {
46 const auto input_index = cur_op->inputs()->operator[](0);
47 const auto filter_index = cur_op->inputs()->operator[](1);
48 const auto bias_index = cur_op->inputs()->operator[](2);
49 const auto output_index = cur_op->outputs()->operator[](0);
50
51 assert(input_index != -1);
52 assert(filter_index != -1);
53 assert(output_index != -1);
54
55 _input_tensor = _runtime_graph->getCircleTensorByIndex(input_index);
56 _filter_tensor = _runtime_graph->getCircleTensorByIndex(filter_index);
57 _bias_tensor = _runtime_graph->getCircleTensorByIndex(bias_index);
58 _output_tensor = _runtime_graph->getCircleTensorByIndex(output_index);
59
60 assert(_input_tensor != nullptr);
61 assert(_filter_tensor != nullptr);
62 }
const circle::Tensor * getCircleTensorByIndex(int32_t index)

References luci_interpreter::RuntimeGraph::getCircleTensorByIndex().

Member Function Documentation

◆ bias()

const circle::Tensor * luci_interpreter::kernels::DownsamplingConv2DKernel::bias ( ) const
inline

◆ filter()

const circle::Tensor * luci_interpreter::kernels::DownsamplingConv2DKernel::filter ( ) const
inline

◆ input()

const circle::Tensor * luci_interpreter::kernels::DownsamplingConv2DKernel::input ( ) const
inline

◆ output()

const circle::Tensor * luci_interpreter::kernels::DownsamplingConv2DKernel::output ( ) const
inline

◆ runtime_graph()

BaseRuntimeGraph * luci_interpreter::kernels::DownsamplingConv2DKernel::runtime_graph ( ) const
inline

Definition at line 69 of file ConvolutionCommon.h.

69{ return _runtime_graph; }

The documentation for this class was generated from the following file: