24.08
|
#include <Layer.hpp>
#include <armnn/Tensor.hpp>
#include <armnn/Types.hpp>
#include "common/include/ProfilingGuid.hpp"
#include <tosa_serialization_handler.h>
Go to the source code of this file.
Functions | |
DType | ArmNNToDType (const DataType &type) |
DataType | DtypeToArmNN (const DType type) |
std::vector< int32_t > | GetTosaTensorShape (const TensorShape &shape) |
std::string | GenerateUniqueInputName (const armnn::InputSlot &slot) |
std::string | GenerateUniqueOutputName (const Layer &layer, uint32_t layerSlot=0) |
std::string | GetUniqueTosaMappingID () |
std::string | TosaDTypeToString (DType tosaDType) |
std::string | TosaOpToString (Op tosaOp) |
std::vector< uint8_t > | ConvertConstantTensorDataToBuffer (const std::shared_ptr< ConstTensorHandle > &tensorHandle) |
std::vector< uint8_t > | CreateConstTosaData (const void *value, DType dtype, const std::vector< int32_t > &shape) |
template<typename T > | |
void | CreateConstTosaOperator (const std::string &outputName, const T value, DType dtype, const std::vector< int32_t > &shape, TosaSerializationOperator *&op, TosaSerializationTensor *&tensor) |
Variables | |
const std::string | mainName = "main" |
|
inline |
Definition at line 22 of file TosaOperatorUtils.hpp.
References armnn::BFloat16, armnn::Boolean, armnn::Float16, armnn::Float32, armnn::QAsymmS8, armnn::QAsymmU8, armnn::QSymmS16, armnn::QSymmS8, armnn::Signed32, and armnn::Signed64.
Referenced by ConvertAvgPool2DIgnoreValueToTosaOperator(), ConvertBatchMatMulToTosaOperator(), ConvertConv2dToTosaOperator(), ConvertDepthwiseConv2dToTosaOperator(), ConvertElementwiseBinaryToTosaOperator(), ConvertFullyConnectedToTosaOperator(), ConvertPooling2DToTosaOperator(), ConvertQuantizeToTosaOperator(), and ConvertSoftmaxToTosaOperator().
|
inline |
Definition at line 333 of file TosaOperatorUtils.hpp.
References armnn::error, armnn::Float16, armnn::Float32, armnn::QAsymmS8, armnn::QAsymmU8, armnn::QSymmS16, armnn::QSymmS8, and armnn::Signed32.
Referenced by ConvertConstantToTosaOperator().
|
inline |
Definition at line 402 of file TosaOperatorUtils.hpp.
References armnn::error.
Referenced by CreateConstTosaOperator().
|
inline |
Definition at line 490 of file TosaOperatorUtils.hpp.
References ARMNN_THROW_MSG_IF_FALSE, and CreateConstTosaData().
|
inline |
Definition at line 52 of file TosaOperatorUtils.hpp.
References armnn::BFloat16, armnn::Boolean, armnn::Float16, armnn::Float32, armnn::QAsymmU8, armnn::QSymmS16, armnn::QSymmS8, and armnn::Signed32.
|
inline |
Definition at line 109 of file TosaOperatorUtils.hpp.
References OutputSlot::CalculateIndexOnOwner(), InputSlot::GetConnectedOutputSlot(), and OutputSlot::GetOwningLayer().
Referenced by ConvertAvgPool2DIgnoreValueToTosaOperator(), ConvertBatchMatMulToTosaOperator(), ConvertConcatToTosaOperator(), ConvertConv2dToTosaOperator(), ConvertDepthwiseConv2dToTosaOperator(), ConvertElementwiseBinaryToTosaOperator(), ConvertExpOperator(), ConvertFullyConnectedToTosaOperator(), ConvertGeluToTosaOperator(), ConvertLeakyReluToTosaOperator(), ConvertLogOperator(), ConvertPadToTosaOperator(), ConvertPooling2DToTosaOperator(), ConvertQuantizeToTosaOperator(), ConvertReluToTosaOperator(), ConvertReshapeToTosaOperator(), ConvertResizeToTosaOperator(), ConvertRsqrtOperator(), ConvertSliceToTosaOperator(), ConvertSoftmaxToTosaOperator(), ConvertTransposeConv2dToTosaOperator(), and ConvertTransposeToTosaOperator().
|
inline |
Definition at line 120 of file TosaOperatorUtils.hpp.
References OutputSlot::GetConnection(), Layer::GetOutputSlot(), InputSlot::GetOwningLayer(), Layer::GetType(), and armnn::Output.
Referenced by ConvertAvgPool2DIgnoreValueToTosaOperator(), ConvertBatchMatMulToTosaOperator(), ConvertConcatToTosaOperator(), ConvertConv2dToTosaOperator(), ConvertDepthwiseConv2dToTosaOperator(), ConvertElementwiseBinaryToTosaOperator(), ConvertExpOperator(), ConvertFullyConnectedToTosaOperator(), ConvertGeluToTosaOperator(), ConvertLeakyReluToTosaOperator(), ConvertLogOperator(), ConvertPadToTosaOperator(), ConvertPooling2DToTosaOperator(), ConvertQuantizeToTosaOperator(), ConvertReluToTosaOperator(), ConvertReshapeToTosaOperator(), ConvertResizeToTosaOperator(), ConvertRsqrtOperator(), ConvertSliceToTosaOperator(), ConvertSoftmaxToTosaOperator(), ConvertTransposeConv2dToTosaOperator(), and ConvertTransposeToTosaOperator().
|
inline |
Definition at line 79 of file TosaOperatorUtils.hpp.
References TensorShape::GetNumDimensions().
Referenced by AddRescaleOp(), ConvertAvgPool2DIgnoreValueToTosaOperator(), ConvertBatchMatMulToTosaOperator(), ConvertConcatToTosaOperator(), ConvertConv2dToTosaOperator(), ConvertDepthwiseConv2dToTosaOperator(), ConvertElementwiseBinaryToTosaOperator(), ConvertExpOperator(), ConvertFullyConnectedToTosaOperator(), ConvertGeluToTosaOperator(), ConvertLeakyReluToTosaOperator(), ConvertLogOperator(), ConvertPadToTosaOperator(), ConvertPooling2DToTosaOperator(), ConvertQuantizeToTosaOperator(), ConvertReduceToTosaOperator(), ConvertReluToTosaOperator(), ConvertReshapeToTosaOperator(), ConvertResizeToTosaOperator(), ConvertRsqrtOperator(), ConvertSliceToTosaOperator(), ConvertSoftmaxToTosaOperator(), ConvertTransposeConv2dToTosaOperator(), and ConvertTransposeToTosaOperator().
|
inline |
Definition at line 138 of file TosaOperatorUtils.hpp.
Referenced by ConvertAvgPool2DIgnoreValueToTosaOperator(), ConvertBatchMatMulToTosaOperator(), ConvertConcatToTosaOperator(), ConvertConstantToTosaOperator(), ConvertConv2dToTosaOperator(), ConvertDepthwiseConv2dToTosaOperator(), ConvertElementwiseBinaryToTosaOperator(), ConvertExpOperator(), ConvertFullyConnectedToTosaOperator(), ConvertGeluToTosaOperator(), ConvertLeakyReluToTosaOperator(), ConvertLogOperator(), ConvertPadToTosaOperator(), ConvertPooling2DToTosaOperator(), ConvertQuantizeToTosaOperator(), ConvertReduceToTosaOperator(), ConvertReluToTosaOperator(), ConvertReshapeToTosaOperator(), ConvertResizeToTosaOperator(), ConvertRsqrtOperator(), ConvertSliceToTosaOperator(), ConvertSoftmaxToTosaOperator(), ConvertTransposeConv2dToTosaOperator(), and ConvertTransposeToTosaOperator().
|
inline |
Definition at line 144 of file TosaOperatorUtils.hpp.
|
inline |
Definition at line 179 of file TosaOperatorUtils.hpp.
const std::string mainName = "main" |
Definition at line 19 of file TosaOperatorUtils.hpp.
Referenced by ConvertQuantizeToTosaOperator(), and ConvertSoftmaxToTosaOperator().