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 ValidationOutputAccessor::access_tensor(), SubTensor::buffer(), NEMaxUnpoolingLayerKernel::configure(), ITensor::copy_from(), CpuAuxTensorHandler::CpuAuxTensorHandler(), Accessor::data(), DetectionOutputAccessor::DetectionOutputAccessor(), arm_compute::utils::draw_detection_rectangle(), Iterator::Iterator(), arm_compute::cpu::nearest_neon_scale(), CpuDepthwiseConv2dAssemblyDispatch::prepare(), ITensor::ptr_to_element(), RandomAccessor::RandomAccessor(), CPPUpsampleKernel::run(), NEROIPoolingLayerKernel::run(), NEReorgLayerKernel::run(), NEROIAlignLayerKernel::run(), CpuGemmConvolution::run(), CpuConcatenateHeightKernel::run_op(), CpuConcatenateWidthKernel::run_op(), CpuDepthwiseConv2dAssemblyWrapperKernel::run_op(), CpuPool2dAssemblyWrapperKernel::run_op(), CpuWinogradConv2dTransformInputKernel< T, OutputTileRows, OutputTileCols, KernelRows, KernelCols >::run_op(), CpuWinogradConv2dTransformOutputKernel< T, OutputTileRows, OutputTileCols, KernelRows, KernelCols >::run_op(), CpuWinogradConv2dTransformWeightsKernel< T, OutputTileRows, OutputTileCols, KernelRows, KernelCols >::run_op(), arm_compute::run_reverse(), TopNPredictionsAccessor::TopNPredictionsAccessor(), NEFFTDigitReverseKernel::validate(), and NEBoundingBoxTransformKernel::validate().

◆ 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 35 of file ITensor.cpp.

References IOFormatInfo::align_columns, ValidRegion::anchor, ARM_COMPUTE_ERROR_ON, BorderSize::bottom, ITensor::buffer(), ITensorInfo::data_type(), IOFormatInfo::Default, ITensorInfo::dimension(), Window::DimY, dt, IOFormatInfo::element_delim, ITensorInfo::element_size(), arm_compute::execute_window_loop(), IOFormatInfo::Full, ITensor::info(), arm_compute::is_data_type_float(), BorderSize::left, IOFormatInfo::NoPadding, ITensorInfo::num_channels(), ITensorInfo::num_dimensions(), offset(), ITensorInfo::offset_element_in_bytes(), ITensorInfo::offset_first_element_in_bytes(), ITensorInfo::padding(), IOFormatInfo::precision, IOFormatInfo::precision_type, ITensor::print(), IOFormatInfo::print_region, Iterator::ptr(), BorderSize::right, IOFormatInfo::row_delim, ITensorInfo::set_valid_region(), ValidRegion::shape, arm_compute::test::validation::src_info, ITensorInfo::strides_in_bytes(), ITensorInfo::tensor_shape(), BorderSize::top, TensorShape::total_size_upper(), Window::use_tensor_dimensions(), arm_compute::test::validation::valid_region, ITensorInfo::valid_region(), IOFormatInfo::ValidRegion, Dimensions< T >::x(), and Dimensions< T >::y().

Referenced by ITensor::ptr_to_element().

 {
     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);
 }

◆ 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 163 of file ITensor.cpp.

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

 {
     return _is_used;
 }

◆ mark_as_unused()

void mark_as_unused ( ) const

Marks a tensor as unused.

Definition at line 168 of file ITensor.cpp.

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

 {
     _is_used = false;
 }

◆ mark_as_used()

void mark_as_used ( ) const

Marks a tensor as used.

Definition at line 173 of file ITensor.cpp.

Referenced by ITensor::ptr_to_element().

 {
     _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 NumPyAccessor::access_tensor(), ITensor::copy_from(), arm_compute::graph::execute_task(), main(), and ITensor::ptr_to_element().

◆ 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.

References ITensor::buffer(), ITensor::copy_from(), ITensor::info(), ITensor::is_used(), ITensor::mark_as_unused(), ITensor::mark_as_used(), ITensorInfo::offset_element_in_bytes(), ITensor::print(), and arm_compute::test::validation::src.

Referenced by arm_compute::cpu::bilinear_neon_scale(), CpuScaleKernel::configure(), NECropKernel::configure_output_shape(), arm_compute::test::validation::DATA_TEST_CASE(), arm_compute::cpu::nearest_neon_scale(), NEGatherKernel::NEGatherKernel(), Accessor::operator()(), CLAccessor::operator()(), arm_compute::cpu::poolingMxN_q8_neon_nhwc(), ClGemmLowpMatrixMultiplyCore::prepare(), CLDepthwiseConvolutionLayer::prepare(), CaffePreproccessor::preprocess(), TFPreproccessor::preprocess(), RandomAccessor::RandomAccessor(), arm_compute::roi_align_1x1(), arm_compute::roi_align_1x1_qasymm8(), CPPDetectionOutputLayer::run(), NETileKernel::run(), NESpaceToDepthLayerKernel::run(), NEDepthToSpaceLayerKernel::run(), CPPNonMaximumSuppressionKernel::run(), CPPDetectionPostProcessLayer::run(), NEROIAlignLayerKernel::run(), NEBatchToSpaceLayerKernel::run(), NECropKernel::run(), NESpaceToBatchLayerKernel::run(), NECropResize::run(), CPPBoxWithNonMaximaSuppressionLimitKernel::run_nmslimit(), NEFillBorderKernel::run_op(), arm_compute::run_reverse(), and NEComputeAllAnchorsKernel::validate().

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

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()