ONE - On-device Neural Engine
Loading...
Searching...
No Matches
arm_compute Namespace Reference

Namespaces

namespace  misc
 
namespace  weights_transformations
 

Data Structures

class  CLBinaryLogicalOp
 
class  CLBinaryLogicalOpKernel
 
class  CLCastBool
 Class to run CLCastBoolKernel. This converts the boolean input tensor to the output tensor's type. More...
 
class  CLCastBoolKernel
 Class for the kernel converting boolean type. More...
 
class  CLDirectTransposeConvLayer
 
class  CLEmbeddingLookup
 Class to perform EmbeddingLookup operation. More...
 
class  CLEmbeddingLookupKernel
 Class to perform EmbeddingLookup operation with opencl kernel. More...
 
class  CLFullyConnectedHybridLayer
 
class  CLFullyConnectedLayerEx
 
class  CLFullyConnectedReshapingLayer
 Class to run FullyConnected Layer after reshaping input tensor. More...
 
class  CLGatherEx
 Class to to run CLGatherKernel. More...
 
class  CLGatherExKernel
 Class to define an interface for the gather kernel. More...
 
class  CLGEMMMatrixAccumulateBiasesKernel
 
class  CLHashtableLookup
 Class to perform HashtableLookup operation. More...
 
class  CLHashtableLookupKernel
 Class to perform HashtableLookup operation with opencl kernel. More...
 
class  CLInstanceNormalizationLayerEx
 
class  CLInstanceNormalizationLayerKernelEx
 
class  CLKernelLibraryEx
 Class to build OpenCL kernels added from nnfw. More...
 
class  CLMemsetKernel
 
class  CLMultiplyScaleFactorKernel
 
class  CLNeg
 
class  CLNegKernel
 
class  CLOneHot
 
class  CLOneHotKernel
 
class  CLPadLayerEx
 
class  CLPadLayerKernelEx
 
class  CLQuantizationSymmetricKernel
 
class  CLReduceOperation
 Class to perform ReduceOperation. More...
 
class  CLReduceOperationKernel
 Class to define interface for the reduce operation kernel. More...
 
class  CLScaleFactorSymm8Kernel
 
class  CLSplitVEx
 
class  CLTopKV2
 Class to execute TopKV2 operation. More...
 
class  CLTransposeConvLayer
 
class  NECastBool
 Class to run INESimpleFunctionNoBorder. More...
 
class  NECastBoolKernel
 Class for the kernel converting boolean type. More...
 
class  NEEmbeddingLookup
 Class to perform EmbeddingLookup operation. More...
 
class  NEEmbeddingLookupKernel
 
class  NEFullyConnectedHybridLayer
 
class  NEFullyConnectedLayerEx
 
class  NEFullyConnectedReshapingLayer
 Class to run FullyConnected Layer after reshaping input tensor. More...
 
class  NEGatherEx
 
class  NEGatherKernelEx
 
class  NEGEMMMatrixAccumulateBiasesKernel
 
class  NEHashtableLookup
 Class to perform HashtableLookup operation. More...
 
class  NEHashtableLookupKernel
 
class  NEInstanceNormalizationLayerEx
 
class  NEInstanceNormalizationLayerKernelEx
 
class  NEMultiplyScaleFactorKernel
 
class  NEOneHot
 
class  NEOneHotKernel
 
class  NEQuantizationSymmetricKernel
 
class  NEReduceOperation
 
class  NEReduceSum
 
class  NETransposeConvLayer
 

Enumerations

enum class  ArgOperation { MAX , MIN }
 
enum class  BinaryLogicalOperation { AND , OR }
 
enum class  ComparisonOperationEx { EQUAL , NOT_EQUAL }
 
enum class  ElementWiseUnaryEx { NEG }
 
enum class  SubDataType { NONE , BOOL }
 

Functions

const std::pair< unsigned int, unsigned int > transposeconv_output_dimensions (unsigned int in_width, unsigned int in_height, unsigned int kernel_width, unsigned int kernel_height, const PadStrideInfo &info, unsigned int invalid_right, unsigned int invalid_top)
 

Enumeration Type Documentation

◆ ArgOperation

enum class arm_compute::ArgOperation
strong

Available ArgIndex operations

Enumerator
MAX 
MIN 

Definition at line 48 of file TypesEx.h.

◆ BinaryLogicalOperation

Available binary logical operations

Enumerator
AND 

AND

OR 

OR

Definition at line 55 of file TypesEx.h.

◆ ComparisonOperationEx

Enumerator
EQUAL 

EQUAL

NOT_EQUAL 

NOT_EQUAL

Definition at line 61 of file TypesEx.h.

◆ ElementWiseUnaryEx

Enumerator
NEG 

NEG

Definition at line 67 of file TypesEx.h.

◆ SubDataType

enum class arm_compute::SubDataType
strong
Enumerator
NONE 
BOOL 

Definition at line 72 of file TypesEx.h.

Function Documentation

◆ transposeconv_output_dimensions()

const std::pair< unsigned int, unsigned int > arm_compute::transposeconv_output_dimensions ( unsigned int  in_width,
unsigned int  in_height,
unsigned int  kernel_width,
unsigned int  kernel_height,
const PadStrideInfo &  info,
unsigned int  invalid_right,
unsigned int  invalid_top 
)

Returns expected width and height of the transpose convolution's output tensor.

Note
This function was copied in order to fix a bug computing to wrong output dimensions.
Parameters
[in]in_widthWidth of input tensor (Number of columns)
[in]in_heightHeight of input tensor (Number of rows)
[in]kernel_widthKernel width.
[in]kernel_heightKernel height.
[in]infopadding and stride info.
[in]invalid_rightThe number of zeros added to right edge of the output.
[in]invalid_topThe number of zeros added to bottom edge of the output.
Returns
A pair with the new width in the first position and the new height in the second.

Definition at line 47 of file UtilsEx.cpp.

51{
52 const unsigned int stride_x = info.stride().first;
53 const unsigned int stride_y = info.stride().second;
54 const unsigned int padx = info.pad_left() + info.pad_right();
55 const unsigned int pady = info.pad_top() + info.pad_bottom();
56
57 ARM_COMPUTE_ERROR_ON(in_width < 1 || in_height < 1);
58 ARM_COMPUTE_ERROR_ON(kernel_width <= padx);
59 ARM_COMPUTE_ERROR_ON(kernel_height <= pady);
60
61 // Find the transpose conv out dimensions
62 // transpose conv out:
63 // tconv_out + pad = 1 + (in - 1) * stride + invalid
64 // tconv_out = 1 + (in - 1) * stride + invalid - pad
65 const int w = stride_x * (in_width - 1) + kernel_width - padx + invalid_right;
66 const int h = stride_y * (in_height - 1) + kernel_height - pady + invalid_bottom;
67
68 return std::make_pair<unsigned int, unsigned int>(w, h);
69}
volatile const char info[]

References info.

Referenced by arm_compute::CLDirectTransposeConvLayer::configure(), arm_compute::NETransposeConvLayer::configure(), arm_compute::NETransposeConvLayer::validate(), and arm_compute::CLDirectTransposeConvLayer::validate().