ArmNN
NotReleased
|
#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 |
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 |
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().