Tensor Class Reference

Tensor object. More...

#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...
std::unique_ptr< ITensorAccessor >	extract_accessor ()
	Extracts accessor from the tensor. 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...
std::set< EdgeID >	bound_edges () const
	Accessor the edges that are bound with the tensor. More...

Detailed Description

Tensor object.

Definition at line 41 of file Tensor.h.

Constructor & Destructor Documentation

Tensor()

Tensor	(	TensorID	id,
		TensorDescriptor	desc
	)

Default constructor.

Parameters

[in]	id	Tensor ID
[in]	desc	Tensor information

Definition at line 30 of file Tensor.cpp.

    : _id(id), _desc(std::move(desc)), _handle(nullptr), _accessor(nullptr), _bound_edges()
{
}

Member Function Documentation

accessor()

ITensorAccessor * accessor

(

)

Backend tensor accessor.

Returns

Backend tensor accessor

Definition at line 65 of file Tensor.cpp.

{
    return _accessor.get();
}

Referenced by arm_compute::graph::detail::fuse_pad_with_convolution(), and Tensor::set_accessor().

bind_edge()

void bind_edge

(

EdgeID

eid

)

Binds the tensor with an edge.

Parameters

[in]

eid

Edge ID that is bound to the tensor

Definition at line 109 of file Tensor.cpp.

{
    _bound_edges.insert(eid);
}

Referenced by Graph::add_connection(), and INode::set_output_tensor().

bound_edges()

std::set< EdgeID > bound_edges

(

)

const

Accessor the edges that are bound with the tensor.

Returns

Bound edges

Definition at line 119 of file Tensor.cpp.

{
    return _bound_edges;
}

call_accessor()

bool call_accessor

(

)

Calls accessor on tensor.

Returns

True if the accessor was called else false

Definition at line 75 of file Tensor.cpp.

{
    // Early exit guard
    if(!_accessor || !_handle)
    {
        return false;
    }
 
    const bool access_data = _accessor->access_tensor_data();
 
    if(access_data)
    {
        // Map tensor
        _handle->map(true);
 
        // Return in case of null backend buffer
        if(_handle->tensor().buffer() == nullptr)
        {
            return false;
        }
    }
 
    // Call accessor
    bool retval = _accessor->access_tensor(_handle->tensor());
 
    if(access_data)
    {
        // Unmap tensor
        _handle->unmap();
    }
 
    return retval;
}

desc() [1/2]

const TensorDescriptor & desc

(

)

TensorInfo metadata accessor.

Returns

Tensor descriptor metadata

Definition at line 40 of file Tensor.cpp.

{
    return _desc;
}

Referenced by BoundingBoxTransformLayerNode::configure_output(), ROIAlignLayerNode::configure_output(), DetectionOutputLayerNode::configure_output(), PriorBoxLayerNode::configure_output(), DeconvolutionLayerNode::configure_output(), EltwiseLayerNode::configure_output(), DepthwiseConvolutionLayerNode::configure_output(), FusedConvolutionBatchNormalizationWithPostOpsNode::configure_output(), ConvolutionLayerNode::configure_output(), FusedConvolutionWithPostOpNode::configure_output(), FusedDepthwiseConvolutionBatchNormalizationNode::configure_output(), FusedConvolutionBatchNormalizationNode::configure_output(), arm_compute::graph::backends::detail::create_concatenate_layer(), SplitLayerNode::forward_descriptors(), arm_compute::graph::detail::fuse_convolution_batch_normalization_with_post_ops(), arm_compute::graph::detail::fuse_convolution_with_post_ops(), arm_compute::graph::detail::fuse_pad_with_convolution(), and SplitLayerSubTensorMutator::mutate().

desc() [2/2]

const TensorDescriptor& desc

(

)

const

TensorInfo metadata accessor.

Returns

Tensor descriptor metadata

extract_accessor()

std::unique_ptr< ITensorAccessor > extract_accessor

(

)

Extracts accessor from the tensor.

Warning

Accessor gets unbound from the tensor

Returns

The accessor of the tensor

Definition at line 70 of file Tensor.cpp.

{
    return std::move(_accessor);
}

Referenced by arm_compute::graph::detail::transfer_driving_nodes_and_remove_old_node().

handle()

ITensorHandle * handle

(

)

Backend tensor handle accessor.

Returns

Backend tensor handle

Definition at line 55 of file Tensor.cpp.

{
    return _handle.get();
}

Referenced by SplitLayerSubTensorMutator::mutate().

id()

TensorID id

(

)

const

Tensor ID accessor.

Returns

Tensor ID

Definition at line 35 of file Tensor.cpp.

{
    return _id;
}

Referenced by Edge::tensor_id().

set_accessor()

void set_accessor

(

std::unique_ptr< ITensorAccessor >

accessor

)

Sets the backend tensor accessor.

Parameters

[in]

accessor

Accessor to set

Definition at line 60 of file Tensor.cpp.

{
    _accessor = std::move(accessor);
}

References Tensor::accessor().

Referenced by GroupedConvolutionMutator::mutate(), and arm_compute::graph::detail::transfer_driving_nodes_and_remove_old_node().

set_handle()

void set_handle

(

std::unique_ptr< ITensorHandle >

backend_tensor

)

Sets the backend tensor.

Parameters

[in]

backend_tensor

Backend tensor to set

Definition at line 50 of file Tensor.cpp.

{
    _handle = std::move(backend_tensor);
}

Referenced by SplitLayerSubTensorMutator::mutate().

unbind_edge()

void unbind_edge

(

EdgeID

eid

)

Unbinds an edge from a tensor.

Parameters

[in]

eid

Edge to unbind

Definition at line 114 of file Tensor.cpp.

{
    _bound_edges.erase(eid);
}

Referenced by Graph::remove_connection(), and INode::set_output_tensor().

---

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

• arm_compute/graph/Tensor.h
• src/graph/Tensor.cpp