nnkit::support::tflite Namespace Reference

Data Structures
class	AbstractBackend
class	InputTensorSet
class	OutputTensorSet
class	TensorContext
struct	TensorSet

Functions
nncc::core::ADT::tensor::Shape	tensor_shape (const TfLiteTensor *t)

Function Documentation

tensor_shape()

nncc::core::ADT::tensor::Shape nnkit::support::tflite::tensor_shape

(

const TfLiteTensor *

t

)

Definition at line 26 of file TensorUtils.cpp.

{
  nncc::core::ADT::tensor::Shape shape;
 
  const int rank = t->dims->size;
 
  shape.resize(rank);
  for (int axis = 0; axis < rank; ++axis)
  {
    shape.dim(axis) = t->dims->data[axis];
  }
 
  return shape;
}

References nncc::core::ADT::tensor::Shape::dim(), and nncc::core::ADT::tensor::Shape::resize().

Referenced by nnkit::support::tflite::TensorContext::shape().