#include "TosaOperatorUtils.hpp"
#include <Layer.hpp>
#include <tosa_serialization_handler.h>
Go to the source code of this file.
◆ ConvertConstantToTosaOperator()
TosaSerializationBasicBlock* ConvertConstantToTosaOperator |
( |
const Layer * |
layer, |
|
|
const std::vector< const TensorInfo * > & |
outputs |
|
) |
| |
Definition at line 10 of file ConstantOperator.cpp.
13 std::string outputName = std::string(
"constant_");
16 std::vector<uint8_t> uint8Data;
22 outputName.append(std::to_string(layer->
GetGuid()));
23 blockName.append(std::to_string(layer->
GetGuid()));
25 auto constantLayer = PolymorphicDowncast<const armnn::ConstantLayer*>(layer);
26 auto tensorInfo = constantLayer->GetOutputSlot().GetTensorInfo();
31 auto* op =
new TosaSerializationOperator(Op_CONST, Attribute_NONE,
nullptr, {}, {outputName});
34 DType outputDType0 =
ArmNNToDType(outputs[0]->GetDataType());
37 auto* outputTensor0 =
new TosaSerializationTensor(outputName, outputShape0, outputDType0, uint8Data);
39 return new TosaSerializationBasicBlock(blockName,
References ConvertConstantTensorDataToBuffer(), Layer::GetGuid(), GetTosaTensorShape(), and GetUniqueTosaMappingID().
Referenced by GetTosaMapping().