23.11
|
#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) |
std::vector< int32_t > | GetTosaTensorShape (const TensorShape &shape) |
std::string | GenerateUniqueName (const Layer &layer, uint32_t layerSlot) |
std::string | GenerateUniqueOutputName (const Layer &layer, uint32_t layerSlot) |
std::string | GetUniqueTosaMappingID () |
std::string | TosaDTypeToString (DType tosaDType) |
std::string | TosaOpToString (Op tosaOp) |
std::vector< uint8_t > | ConvertConstantTensorDataToBuffer (const std::shared_ptr< ConstTensorHandle > &tensorHandle) |
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(), and ConvertPooling2DToTosaOperator().
|
inline |
Definition at line 289 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 63 of file TosaOperatorUtils.hpp.
References armnn::Constant, Layer::GetGuid(), Layer::GetType(), armnn::Input, and armnn::Output.
Referenced by ConvertAvgPool2DIgnoreValueToTosaOperator(), ConvertConcatToTosaOperator(), ConvertConv2dToTosaOperator(), ConvertElementwiseBinaryToTosaOperator(), ConvertElementwiseUnaryOperator(), ConvertPooling2DToTosaOperator(), ConvertReshapeToTosaOperator(), ConvertSliceToTosaOperator(), ConvertTransposeConv2dToTosaOperator(), ConvertTransposeToTosaOperator(), and GenerateUniqueOutputName().
|
inline |
Definition at line 82 of file TosaOperatorUtils.hpp.
References GenerateUniqueName(), OutputSlot::GetConnection(), Layer::GetOutputSlot(), InputSlot::GetOwningLayer(), Layer::GetType(), and armnn::Output.
Referenced by ConvertAvgPool2DIgnoreValueToTosaOperator(), ConvertConcatToTosaOperator(), ConvertConv2dToTosaOperator(), ConvertElementwiseBinaryToTosaOperator(), ConvertElementwiseUnaryOperator(), ConvertPooling2DToTosaOperator(), ConvertReshapeToTosaOperator(), ConvertSliceToTosaOperator(), ConvertTransposeConv2dToTosaOperator(), and ConvertTransposeToTosaOperator().
|
inline |
Definition at line 52 of file TosaOperatorUtils.hpp.
References TensorShape::GetNumDimensions().
Referenced by ConvertAvgPool2DIgnoreValueToTosaOperator(), ConvertConcatToTosaOperator(), ConvertConstantToTosaOperator(), ConvertConv2dToTosaOperator(), ConvertElementwiseBinaryToTosaOperator(), ConvertElementwiseUnaryOperator(), ConvertPooling2DToTosaOperator(), ConvertReshapeToTosaOperator(), ConvertSliceToTosaOperator(), ConvertTransposeConv2dToTosaOperator(), and ConvertTransposeToTosaOperator().
|
inline |
Definition at line 100 of file TosaOperatorUtils.hpp.
Referenced by ConvertAvgPool2DIgnoreValueToTosaOperator(), ConvertConcatToTosaOperator(), ConvertConstantToTosaOperator(), ConvertConv2dToTosaOperator(), ConvertElementwiseBinaryToTosaOperator(), ConvertElementwiseUnaryOperator(), ConvertPooling2DToTosaOperator(), ConvertReshapeToTosaOperator(), ConvertSliceToTosaOperator(), ConvertTransposeConv2dToTosaOperator(), and ConvertTransposeToTosaOperator().
|
inline |
Definition at line 106 of file TosaOperatorUtils.hpp.
|
inline |
Definition at line 139 of file TosaOperatorUtils.hpp.
const std::string mainName = "main" |
Definition at line 19 of file TosaOperatorUtils.hpp.