20.02
|
Provides access to the appropriate indexes for Channels, Height and Width based on DataLayout. More...
#include <DataLayoutIndexed.hpp>
Public Member Functions | |
DataLayoutIndexed (armnn::DataLayout dataLayout) | |
armnn::DataLayout | GetDataLayout () const |
unsigned int | GetChannelsIndex () const |
unsigned int | GetHeightIndex () const |
unsigned int | GetWidthIndex () const |
unsigned int | GetIndex (const armnn::TensorShape &shape, unsigned int batchIndex, unsigned int channelIndex, unsigned int heightIndex, unsigned int widthIndex) const |
Provides access to the appropriate indexes for Channels, Height and Width based on DataLayout.
Definition at line 17 of file DataLayoutIndexed.hpp.
DataLayoutIndexed | ( | armnn::DataLayout | dataLayout | ) |
Definition at line 13 of file DataLayoutIndexed.cpp.
References armnn::NCHW, and armnn::NHWC.
|
inline |
Definition at line 23 of file DataLayoutIndexed.hpp.
Referenced by armnn::BatchNormImpl(), armnn::BatchToSpaceNd(), armnn::Convolve(), armnn::DepthToSpace(), GetBias(), armnn::GetOffset(), DepthToSpaceLayer::InferOutputShapes(), Pooling2dLayer::InferOutputShapes(), ResizeLayer::InferOutputShapes(), armnn::InstanceNorm(), armnn::Offset(), armnn::Pooling2d(), armnn::Resize(), armnn::SpaceToBatchNd(), armnn::SpaceToDepth(), armnn::TransposeConvolution2dImpl(), ResizeBilinearQueueDescriptor::Validate(), ResizeQueueDescriptor::Validate(), SpaceToBatchNdQueueDescriptor::Validate(), SpaceToDepthQueueDescriptor::Validate(), and DepthToSpaceQueueDescriptor::Validate().
|
inline |
Definition at line 22 of file DataLayoutIndexed.hpp.
Referenced by CompareDepthwiseConvolution2dTestImpl(), armnn::GetOffset(), armnn::Offset(), and armnnUtils::operator==().
|
inline |
Definition at line 24 of file DataLayoutIndexed.hpp.
Referenced by armnn::BatchNormImpl(), armnn::BatchToSpaceNd(), armnnTfParser::ConvertTfTensorDataType(), armnn::Convolve(), armnn::DepthToSpace(), armnn::GetOffset(), SpaceToBatchNdLayer::InferOutputShapes(), DepthToSpaceLayer::InferOutputShapes(), Pooling2dLayer::InferOutputShapes(), BatchToSpaceNdLayer::InferOutputShapes(), armnn::InstanceNorm(), armnn::Offset(), armnnTfParser::OutputShapeOfSqueeze(), armnn::Pooling2d(), armnn::Resize(), Convolution2dLayer::SerializeLayerParameters(), armnn::SpaceToBatchNd(), armnn::SpaceToDepth(), armnn::TransposeConvolution2dImpl(), SpaceToBatchNdQueueDescriptor::Validate(), SpaceToDepthQueueDescriptor::Validate(), and DepthToSpaceQueueDescriptor::Validate().
|
inline |
Offset the given indices appropriately depending on the data layout
channelIndex stays unchanged
widthIndex stays unchanged
Get the value using the correct offset
Definition at line 27 of file DataLayoutIndexed.hpp.
References armnn::NCHW, armnn::NHWC, and armnnUtils::operator==().
Referenced by armnn::BatchNormImpl(), armnn::Convolve(), TensorBufferArrayView< DataType >::Get(), armnn::InstanceNorm(), armnn::Pooling2d(), armnn::Resize(), and armnn::TransposeConvolution2dImpl().
|
inline |
Definition at line 25 of file DataLayoutIndexed.hpp.
Referenced by armnn::BatchNormImpl(), armnn::BatchToSpaceNd(), armnnTfParser::ConvertTfTensorDataType(), armnn::Convolve(), armnn::DepthToSpace(), armnn::GetOffset(), SpaceToBatchNdLayer::InferOutputShapes(), DepthToSpaceLayer::InferOutputShapes(), Pooling2dLayer::InferOutputShapes(), BatchToSpaceNdLayer::InferOutputShapes(), armnn::InstanceNorm(), armnn::Offset(), armnnTfParser::OutputShapeOfSqueeze(), armnn::Pooling2d(), armnn::Resize(), Convolution2dLayer::SerializeLayerParameters(), armnn::SpaceToBatchNd(), armnn::SpaceToDepth(), armnn::TransposeConvolution2dImpl(), SpaceToBatchNdQueueDescriptor::Validate(), SpaceToDepthQueueDescriptor::Validate(), and DepthToSpaceQueueDescriptor::Validate().