luci_interpreter::kernels::VectorOfTensors< T, is_const > Class Template Reference

#include <Utils.h>

Public Types
using	ElementT = typename std::conditional< is_const, const T, T >::type
using	TensorT = typename std::conditional< is_const, const Tensor, Tensor >::type
using	ElementT = typename std::conditional< is_const, const T, T >::type
using	TensorT = typename std::conditional< is_const, const Tensor, Tensor >::type

Public Member Functions
	VectorOfTensors (const std::vector< TensorT * > &tensor_list)
ElementT *const *	data () const
const tflite::RuntimeShape *const *	shapes () const
	VectorOfTensors (const std::vector< TensorT * > &tensor_list)
ElementT *const *	data () const
const luci_interpreter::RuntimeShape *const *	shapes () const

Detailed Description

template<typename T, bool is_const>
class luci_interpreter::kernels::VectorOfTensors< T, is_const >

Definition at line 220 of file Utils.h.

Member Typedef Documentation

ElementT [1/2]

template<typename T , bool is_const>

using luci_interpreter::kernels::VectorOfTensors< T, is_const >::ElementT = typename std::conditional<is_const, const T, T>::type

Definition at line 223 of file Utils.h.

ElementT [2/2]

template<typename T , bool is_const>

using luci_interpreter::kernels::VectorOfTensors< T, is_const >::ElementT = typename std::conditional<is_const, const T, T>::type

Definition at line 176 of file Utils.h.

TensorT [1/2]

template<typename T , bool is_const>

using luci_interpreter::kernels::VectorOfTensors< T, is_const >::TensorT = typename std::conditional<is_const, const Tensor, Tensor>::type

Definition at line 224 of file Utils.h.

TensorT [2/2]

template<typename T , bool is_const>

using luci_interpreter::kernels::VectorOfTensors< T, is_const >::TensorT = typename std::conditional<is_const, const Tensor, Tensor>::type

Definition at line 177 of file Utils.h.

Constructor & Destructor Documentation

VectorOfTensors() [1/2]

template<typename T , bool is_const>

luci_interpreter::kernels::VectorOfTensors< T, is_const >::VectorOfTensors ( const std::vector< TensorT * > &  tensor_list )

inlineexplicit

Definition at line 227 of file Utils.h.

  {
    const int num_tensors = tensor_list.size();
 
    all_data_.reserve(num_tensors);
    all_shape_.reserve(num_tensors);
    all_shape_ptr_.reserve(num_tensors);
 
    for (TensorT *tensor : tensor_list)
    {
      all_data_.push_back(getTensorData<T>(tensor));
      all_shape_.push_back(getTensorShape(tensor));
    }
 
    // Taking the pointer from inside a std::vector is only OK if the vector is
    // never modified, so we populate all_shape in the previous loop and then we
    // are free to grab iterators here.
    for (tflite::RuntimeShape &shape : all_shape_)
    {
      all_shape_ptr_.push_back(&shape);
    }
  }

References luci_interpreter::kernels::getTensorShape().

VectorOfTensors() [2/2]

template<typename T , bool is_const>

luci_interpreter::kernels::VectorOfTensors< T, is_const >::VectorOfTensors ( const std::vector< TensorT * > &  tensor_list )

inlineexplicit

Definition at line 180 of file Utils.h.

  {
    const int num_tensors = tensor_list.size();
 
    all_data_.reserve(num_tensors);
    all_shape_.reserve(num_tensors);
    all_shape_ptr_.reserve(num_tensors);
 
    for (TensorT *tensor : tensor_list)
    {
      all_data_.push_back(getTensorData<T>(tensor));
      all_shape_.push_back(getTensorShape(tensor));
    }
 
    // Taking the pointer from inside a std::vector is only OK if the vector is
    // never modified, so we populate all_shape in the previous loop and then we
    // are free to grab iterators here.
    for (luci_interpreter::RuntimeShape &shape : all_shape_)
    {
      all_shape_ptr_.push_back(&shape);
    }
  }

References luci_interpreter::kernels::getTensorShape().

Member Function Documentation

data() [1/2]

template<typename T , bool is_const>

ElementT *const * luci_interpreter::kernels::VectorOfTensors< T, is_const >::data ( ) const

inline

Definition at line 253 of file Utils.h.

{ return all_data_.data(); }

data() [2/2]

template<typename T , bool is_const>

ElementT *const * luci_interpreter::kernels::VectorOfTensors< T, is_const >::data ( ) const

inline

Definition at line 206 of file Utils.h.

{ return all_data_.data(); }

shapes() [1/2]

template<typename T , bool is_const>

const tflite::RuntimeShape *const * luci_interpreter::kernels::VectorOfTensors< T, is_const >::shapes ( ) const

inline

Definition at line 259 of file Utils.h.

{ return all_shape_ptr_.data(); }

shapes() [2/2]

template<typename T , bool is_const>

const luci_interpreter::RuntimeShape *const * luci_interpreter::kernels::VectorOfTensors< T, is_const >::shapes ( ) const

inline

Definition at line 212 of file Utils.h.

{ return all_shape_ptr_.data(); }

---

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

• compiler/luci-interpreter/src/kernels/Utils.h
• onert-micro/luci-interpreter/src/kernels/Utils.h