ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Public Types | Static Public Member Functions
mir_interpreter::GatherByT< T > Struct Template Reference

Public Types

template<typename IndicesT >
using GatherWithFixedT = GatherImpl< T, IndicesT >
 

Static Public Member Functions

static void run (const TensorVariant &data, const TensorVariant &indices, const ops::GatherOp &op, TensorVariant &res)
 

Detailed Description

template<typename T>
struct mir_interpreter::GatherByT< T >

Definition at line 74 of file Gather.cpp.

Member Typedef Documentation

◆ GatherWithFixedT

template<typename T >
template<typename IndicesT >
using mir_interpreter::GatherByT< T >::GatherWithFixedT = GatherImpl<T, IndicesT>

Definition at line 77 of file Gather.cpp.

Member Function Documentation

◆ run()

template<typename T >
static void mir_interpreter::GatherByT< T >::run ( const TensorVariant &  data,
const TensorVariant &  indices,
const ops::GatherOp &  op,
TensorVariant &  res 
)
inlinestatic

Definition at line 79 of file Gather.cpp.

81 {
82 dispatch<GatherWithFixedT>(indices.getElementType(), data, indices, op, res);
83 }

References mir::TensorVariant::getElementType().

Referenced by package.infer.session::inference().

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