Compute Library
18.05
|
#include <Tensor.h>
Public Member Functions | |
Tensor (TensorID id, TensorDescriptor desc) | |
Default constructor. More... | |
TensorID | id () const |
Tensor ID accessor. More... | |
TensorDescriptor & | desc () |
TensorInfo metadata accessor. More... | |
const TensorDescriptor & | desc () const |
TensorInfo metadata accessor. More... | |
void | set_handle (std::unique_ptr< ITensorHandle > backend_tensor) |
Sets the backend tensor. More... | |
ITensorHandle * | handle () |
Backend tensor handle accessor. More... | |
void | set_accessor (std::unique_ptr< ITensorAccessor > accessor) |
Sets the backend tensor accessor. More... | |
ITensorAccessor * | accessor () |
Backend tensor accessor. More... | |
bool | call_accessor () |
Calls accessor on tensor. More... | |
void | bind_edge (EdgeID eid) |
Binds the tensor with an edge. More... | |
void | unbind_edge (EdgeID eid) |
Unbinds an edge from a tensor. More... | |
const std::set< EdgeID > | bound_edges () const |
Accessor the edges that are bound with the tensor. More... | |
Tensor | ( | TensorID | id, |
TensorDescriptor | desc | ||
) |
ITensorAccessor* accessor | ( | ) |
Backend tensor accessor.
void bind_edge | ( | EdgeID | eid | ) |
Binds the tensor with an edge.
[in] | eid | Edge ID that is bound to the tensor |
const std::set<EdgeID> bound_edges | ( | ) | const |
Accessor the edges that are bound with the tensor.
bool call_accessor | ( | ) |
Calls accessor on tensor.
TensorDescriptor& desc | ( | ) |
TensorInfo metadata accessor.
Referenced by arm_compute::graph::get_tensor_descriptor().
const TensorDescriptor& desc | ( | ) | const |
TensorInfo metadata accessor.
ITensorHandle* handle | ( | ) |
Backend tensor handle accessor.
Referenced by arm_compute::graph::backends::detail::get_backing_tensor_info().
TensorID id | ( | ) | const |
void set_accessor | ( | std::unique_ptr< ITensorAccessor > | accessor | ) |
Sets the backend tensor accessor.
[in] | accessor | Accessor to set |
Referenced by arm_compute::graph::set_tensor_accessor().
void set_handle | ( | std::unique_ptr< ITensorHandle > | backend_tensor | ) |
Sets the backend tensor.
[in] | backend_tensor | Backend tensor to set |