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

#include <MISOKernel.h>

Public Member Functions

 MISOKernel (const circle::Operator *cur_op, BaseRuntimeGraph *runtime_graph)
 
const circle::Tensor * input1 () const
 
const circle::Tensor * input2 () const
 
const circle::Tensor * input3 () const
 
const circle::Tensor * input4 () const
 
const circle::Tensor * output () const
 

Detailed Description

Definition at line 28 of file MISOKernel.h.

Constructor & Destructor Documentation

◆ MISOKernel()

luci_interpreter::kernels::MISOKernel::MISOKernel ( const circle::Operator *  cur_op,
BaseRuntimeGraph runtime_graph 
)
inline

Definition at line 31 of file MISOKernel.h.

32 {
33 const auto input1_index = cur_op->inputs()->operator[](0);
34 const auto input2_index = cur_op->inputs()->operator[](1);
35 const auto input3_index = cur_op->inputs()->operator[](2);
36 const auto input4_index =
37 cur_op->inputs()->size() == 4 ? cur_op->inputs()->operator[](3) : -1; // optional
38
39 const auto output_index = cur_op->outputs()->operator[](0);
40
41 assert(input1_index != -1);
42 assert(input2_index != -1);
43 assert(input3_index != -1);
44
45 assert(output_index != -1);
46
47 _input1_tensor = runtime_graph->getCircleTensorByIndex(input1_index);
48 _input2_tensor = runtime_graph->getCircleTensorByIndex(input2_index);
49 _input3_tensor = runtime_graph->getCircleTensorByIndex(input3_index);
50 _output_tensor = runtime_graph->getCircleTensorByIndex(output_index);
51
52 // optional
53 if (input4_index != -1)
54 _input4_tensor = runtime_graph->getCircleTensorByIndex(input4_index);
55 else
56 _input4_tensor = nullptr;
57
58 assert(_input1_tensor != nullptr);
59 assert(_input2_tensor != nullptr);
60 assert(_input3_tensor != nullptr);
61 assert(_output_tensor != nullptr);
62 }

References luci_interpreter::RuntimeGraph::getCircleTensorByIndex().

Member Function Documentation

◆ input1()

◆ input2()

◆ input3()

◆ input4()

const circle::Tensor * luci_interpreter::kernels::MISOKernel::input4 ( ) const
inline

Definition at line 68 of file MISOKernel.h.

69 {
70 assert(_input4_tensor != nullptr);
71 return _input4_tensor;
72 }

Referenced by luci_interpreter::configure_kernel_CircleStridedSlice(), and luci_interpreter::execute_kernel_CircleStridedSlice().

◆ output()


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