TensorImpl class final
#include <tensor.hpp>
Concrete implementation of the Tensor interface for TensorFlow Lite.
Contents
Base classes
- class edge::Tensor
- A base class for representing a tensor object.
Constructors, destructors, conversion operators
- TensorImpl(TfLiteTensor* tfLiteTensor = nullptr) explicit
- Constructor for TensorImpl.
- TensorImpl(const TensorImpl& other) defaulted
- TensorImpl(TensorImpl&&) defaulted
- ~TensorImpl() defaulted final
Public functions
- auto operator=(const TensorImpl&) -> TensorImpl & defaulted
- auto operator=(TensorImpl&&) -> TensorImpl & defaulted
- auto getName() const -> std::string final
- Get the name of the tensor.
- auto getType() const -> TensorType final
- Get the type of the tensor.
- auto getDimensions() const -> std::vector< size_t > final
- Get the dimensions of the tensor.
- auto getSize() const -> size_t final
- Get the total size of the tensor.
Protected functions
- auto getDataPtr() -> void * final
- Get a pointer to the data of the tensor.
- auto getNumBytes() -> size_t final
- Get the number of bytes occupied by the tensor data.
Function documentation
edge::
Constructor for TensorImpl.
| Parameters | |
|---|---|
| tfLiteTensor | Pointer to the TfLiteTensor object. |
auto edge::
Get the name of the tensor.
| Returns | The name of the tensor as a string. |
|---|
auto edge::
Get the type of the tensor.
| Returns | The type of the tensor as a TensorType enum. |
|---|
auto edge::
Get the dimensions of the tensor.
| Returns | A vector of size_t representing the dimensions of the tensor. |
|---|
auto edge::
Get the total size of the tensor.
| Returns | The total size of the tensor in number of elements. |
|---|
auto edge::
Get a pointer to the data of the tensor.
| Returns | A void pointer to the data of the tensor. |
|---|
auto edge::
Get the number of bytes occupied by the tensor data.
| Returns | The number of bytes occupied by the tensor data. |
|---|