ONE - On-device Neural Engine
Loading...
Searching...
No Matches
luci_interpreter::kernels::VectorOfQuantizedTensors< is_const > Class Template Reference

#include <Utils.h>

Collaboration diagram for luci_interpreter::kernels::VectorOfQuantizedTensors< is_const >:

Public Member Functions

 VectorOfQuantizedTensors (const std::vector< TensorT * > &tensor_list)
 
const float * scale () const
 
const int32_t * zero_point () const
 
 VectorOfQuantizedTensors (const std::vector< TensorT * > &tensor_list)
 
const float * scale () const
 
const int32_t * zero_point () const
 
- Public Member Functions inherited from luci_interpreter::kernels::VectorOfTensors< uint8_t, is_const >
 VectorOfTensors (const std::vector< TensorT * > &tensor_list)
 
 VectorOfTensors (const std::vector< TensorT * > &tensor_list)
 
ElementT *const * data () const
 
ElementT *const * data () const
 
const tflite::RuntimeShape *const * shapes () const
 
const luci_interpreter::RuntimeShape *const * shapes () const
 

Additional Inherited Members

- Public Types inherited from luci_interpreter::kernels::VectorOfTensors< uint8_t, is_const >
using ElementT = typename std::conditional< is_const, const uint8_t, uint8_t >::type
 
using ElementT = typename std::conditional< is_const, const uint8_t, uint8_t >::type
 
using TensorT = typename std::conditional< is_const, const Tensor, Tensor >::type
 
using TensorT = typename std::conditional< is_const, const Tensor, Tensor >::type
 

Detailed Description

template<bool is_const>
class luci_interpreter::kernels::VectorOfQuantizedTensors< is_const >

Definition at line 269 of file Utils.h.

Constructor & Destructor Documentation

◆ VectorOfQuantizedTensors() [1/2]

template<bool is_const>
luci_interpreter::kernels::VectorOfQuantizedTensors< is_const >::VectorOfQuantizedTensors ( const std::vector< TensorT * > &  tensor_list)
inlineexplicit

Definition at line 275 of file Utils.h.

276 : VectorOfTensors<uint8_t, is_const>(tensor_list)
277 {
278 for (TensorT *tensor : tensor_list)
279 {
280 zero_point_.push_back(tensor->zero_point());
281 scale_.push_back(tensor->scale());
282 }
283 }
typename std::conditional< is_const, const Tensor, Tensor >::type TensorT
Definition Utils.h:224

◆ VectorOfQuantizedTensors() [2/2]

template<bool is_const>
luci_interpreter::kernels::VectorOfQuantizedTensors< is_const >::VectorOfQuantizedTensors ( const std::vector< TensorT * > &  tensor_list)
inlineexplicit

Definition at line 316 of file Utils.h.

317 : VectorOfTensors<uint8_t, is_const>(tensor_list)
318 {
319 for (TensorT *tensor : tensor_list)
320 {
321 zero_point_.push_back(tensor->zero_point());
322 scale_.push_back(tensor->scale());
323 }
324 }

Member Function Documentation

◆ scale() [1/2]

template<bool is_const>
const float * luci_interpreter::kernels::VectorOfQuantizedTensors< is_const >::scale ( ) const
inline

Definition at line 285 of file Utils.h.

285{ return scale_.data(); }

◆ scale() [2/2]

template<bool is_const>
const float * luci_interpreter::kernels::VectorOfQuantizedTensors< is_const >::scale ( ) const
inline

Definition at line 326 of file Utils.h.

326{ return scale_.data(); }

◆ zero_point() [1/2]

template<bool is_const>
const int32_t * luci_interpreter::kernels::VectorOfQuantizedTensors< is_const >::zero_point ( ) const
inline

Definition at line 286 of file Utils.h.

286{ return zero_point_.data(); }

◆ zero_point() [2/2]

template<bool is_const>
const int32_t * luci_interpreter::kernels::VectorOfQuantizedTensors< is_const >::zero_point ( ) const
inline

Definition at line 327 of file Utils.h.

327{ return zero_point_.data(); }

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