Interface for CPU tensor. More...

#include <ITensor.h>

Public Member Functions
virtual ITensorInfo *	info () const =0
	Interface to be implemented by the child class to return the tensor's metadata. More...

virtual ITensorInfo *	info ()=0
	Interface to be implemented by the child class to return the tensor's metadata. More...

virtual	~ITensor ()=default
	Default virtual destructor. More...

virtual uint8_t *	buffer () const =0
	Interface to be implemented by the child class to return a pointer to CPU memory. More...

uint8_t *	ptr_to_element (const Coordinates &id) const
	Return a pointer to the element at the passed coordinates. More...

void	copy_from (const ITensor &src)
	Copy the content of another tensor. More...

void	print (std::ostream &s, IOFormatInfo io_fmt=IOFormatInfo()) const
	Print a tensor to a given stream using user defined formatting information. More...

bool	is_used () const
	Flags if the tensor is used or not. More...

void	mark_as_unused () const
	Marks a tensor as unused. More...

void	mark_as_used () const
	Marks a tensor as used. More...

Detailed Description

Interface for CPU tensor.

Definition at line 36 of file ITensor.h.

Constructor & Destructor Documentation

◆ ~ITensor()

virtual ~ITensor ( )

virtualdefault

Default virtual destructor.

Member Function Documentation

◆ buffer()

virtual uint8_t* buffer ( ) const

pure virtual

Interface to be implemented by the child class to return a pointer to CPU memory.

Returns: A CPU pointer to the beginning of the image's allocation.

Implemented in ICLTensor, SubTensor, OperatorTensor, and Tensor.

Referenced by arm_compute::cpu::bounding_box_transform(), arm_compute::cpu::bounding_box_transform_qsymm16(), SubTensor::buffer(), CpuAuxTensorHandler::CpuAuxTensorHandler(), Accessor::data(), arm_compute::cpu::directconv3d_float_neon_ndhwc(), arm_compute::cpu::directconv3d_quantized_neon_ndhwc(), arm_compute::cpu::max_unpooling(), CpuDepthwiseConv2dAssemblyDispatch::prepare(), CpuWinogradConv2d::prepare(), ITensor::ptr_to_element(), arm_compute::cpu::roi_align(), CPPUpsampleKernel::run(), NEROIPoolingLayerKernel::run(), NEReorgLayerKernel::run(), CpuGemmConv2d::run(), CpuWinogradConv2dTransformInputKernel::run_op(), CpuDepthwiseConv2dAssemblyWrapperKernel::run_op(), CpuPool2dAssemblyWrapperKernel::run_op(), CpuWinogradConv2dTransformOutputKernel::run_op(), arm_compute::run_reverse(), and arm_compute::cpu::select_op_not_same_rank().

◆ copy_from()

void copy_from ( const ITensor & src )

Copy the content of another tensor.

Note: The number of dimensions of the source tensor must be less or equal to those of the destination tensor.; All dimensions of the destination tensor must be greater or equal to the source tensor ones.; num_channels() and element_size() of both tensors must match.

Parameters

[in] src Source tensor to copy from.

Definition at line 36 of file ITensor.cpp.

 {
     if(&src == this)
     {
         return;
     }
  
     const ITensorInfo *src_info = src.info();
     ITensorInfo       *dst_info = this->info();
  
     ARM_COMPUTE_ERROR_ON(src_info->num_dimensions() > dst_info->num_dimensions());
     ARM_COMPUTE_ERROR_ON(src_info->num_channels() != dst_info->num_channels());
     ARM_COMPUTE_ERROR_ON(src_info->element_size() != dst_info->element_size());
  
     for(size_t d = 0; d < src_info->num_dimensions(); d++)
     {
         ARM_COMPUTE_ERROR_ON(src_info->dimension(d) > dst_info->dimension(d));
     }
  
     // Copy information about valid region
     dst_info->set_valid_region(src_info->valid_region());
  
     Window win_src;
     win_src.use_tensor_dimensions(src_info->tensor_shape(), Window::DimY);
     Window win_dst;
     win_dst.use_tensor_dimensions(dst_info->tensor_shape(), Window::DimY);
  
     Iterator src_it(&src, win_src);
     Iterator dst_it(this, win_dst);
  
     const size_t line_size = src_info->element_size() * src_info->dimension(0);
  
     execute_window_loop(
         win_src, [&](const Coordinates &)
     {
         memcpy(dst_it.ptr(), src_it.ptr(), line_size);
     },
     src_it, dst_it);
 }

References ARM_COMPUTE_ERROR_ON, ITensorInfo::dimension(), TensorInfo::dimension(), Window::DimY, ITensorInfo::element_size(), TensorInfo::element_size(), arm_compute::execute_window_loop(), ITensor::info(), ITensorInfo::num_channels(), TensorInfo::num_channels(), ITensorInfo::num_dimensions(), TensorInfo::num_dimensions(), Iterator::ptr(), ITensorInfo::set_valid_region(), arm_compute::test::validation::src, arm_compute::test::validation::src_info, ITensorInfo::tensor_shape(), TensorInfo::tensor_shape(), Window::use_tensor_dimensions(), and TensorInfo::valid_region().

◆ info() [1/2]

virtual ITensorInfo* info ( ) const

pure virtual

Interface to be implemented by the child class to return the tensor's metadata.

Returns: A pointer to the tensor's metadata.

Implemented in CLSubTensor, CLTensor, SubTensor, OperatorTensor, and Tensor.

◆ info() [2/2]

virtual ITensorInfo* info ( )

pure virtual

Interface to be implemented by the child class to return the tensor's metadata.

Returns: A pointer to the tensor's metadata.

Implemented in CLSubTensor, CLTensor, SubTensor, OperatorTensor, and Tensor.

◆ is_used()

bool is_used ( ) const

Flags if the tensor is used or not.

Returns: True if it is used else false

Definition at line 164 of file ITensor.cpp.

 {
     return _is_used;
 }

Referenced by CLFullyConnectedLayer::prepare(), CLDepthwiseConvolutionLayer::prepare(), NEFFTConvolutionLayer::prepare(), CLFFTConvolutionLayer::prepare(), NEDeconvolutionLayer::prepare(), CLGEMMDeconvolutionLayer::prepare(), CLDirectDeconvolutionLayer::prepare(), NEFullyConnectedLayer::prepare(), CLTensorHandle::release_if_unused(), and NETensorHandle::release_if_unused().

◆ mark_as_unused()

void mark_as_unused ( ) const

Marks a tensor as unused.

Definition at line 169 of file ITensor.cpp.

 {
     _is_used = false;
 }

Referenced by CpuDepthwiseConv2dAssemblyDispatch::prepare(), ClFullyConnected::prepare(), CpuFullyConnected::prepare(), CLDepthwiseConvolutionLayer::prepare(), NEFFTConvolutionLayer::prepare(), CLFFTConvolutionLayer::prepare(), NEDeconvolutionLayer::prepare(), CLGEMMDeconvolutionLayer::prepare(), NELSTMLayerQuantized::prepare(), CLDirectDeconvolutionLayer::prepare(), CLLSTMLayerQuantized::prepare(), NEQLSTMLayer::prepare(), CLQLSTMLayer::prepare(), IWeightsManager::release(), and IWeightsManager::run().

◆ mark_as_used()

void mark_as_used ( ) const

Marks a tensor as used.

Definition at line 174 of file ITensor.cpp.

 {
     _is_used = true;
 }

◆ print()

void print	(	std::ostream &	s,
		IOFormatInfo	io_fmt = `IOFormatInfo()`
	)		const

Print a tensor to a given stream using user defined formatting information.

Parameters

s	Output stream
io_fmt	Format information

Referenced by main().

◆ ptr_to_element()

uint8_t* ptr_to_element ( const Coordinates & id ) const

inline

Return a pointer to the element at the passed coordinates.

Parameters

[in] id Coordinates of the element

Returns: Pointer to the requested element

Definition at line 63 of file ITensor.h.

     {
         return buffer() + info()->offset_element_in_bytes(id);
     }

References ITensor::buffer(), ITensor::info(), and ITensorInfo::offset_element_in_bytes().

Referenced by arm_compute::cpu::compute_all_anchors(), arm_compute::cpu::compute_all_anchors_qasymm16(), NECropKernel::configure_output_shape(), arm_compute::cpu::fused_batch_normalization_conv(), arm_compute::cpu::fused_batch_normalization_dwc_nchw(), arm_compute::cpu::fused_batch_normalization_dwc_nhwc(), Accessor::operator()(), CLAccessor::operator()(), ClGemmLowpMatrixMultiplyCore::prepare(), CLDepthwiseConvolutionLayer::prepare(), arm_compute::cpu::roi_align(), CPPDetectionOutputLayer::run(), NETileKernel::run(), NEDepthToSpaceLayerKernel::run(), NESpaceToDepthLayerKernel::run(), CPPNonMaximumSuppressionKernel::run(), CPPDetectionPostProcessLayer::run(), NECropKernel::run(), NEBatchToSpaceLayerKernel::run(), NESpaceToBatchLayerKernel::run(), NECropResize::run(), CPPBoxWithNonMaximaSuppressionLimitKernel::run_nmslimit(), and arm_compute::run_reverse().

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

arm_compute/core/ITensor.h
src/core/ITensor.cpp

Public Member Functions

Detailed Description

Constructor & Destructor Documentation

◆ ~ITensor()

Member Function Documentation

◆ buffer()

◆ copy_from()

◆ info() [1/2]

◆ info() [2/2]

◆ is_used()

◆ mark_as_unused()

◆ mark_as_used()

◆ print()

◆ ptr_to_element()