9 const std::vector<const TensorInfo*>& outputs,
22 blockName = std::string(
"main");
25 TosaSerializationOperator* op =
new TosaSerializationOperator(Op_ADD,
28 {input0Name, input1Name},
32 DType inputDType0 =
ArmNNToDType(inputs[0]->GetDataType());
35 DType inputDType1 =
ArmNNToDType(inputs[1]->GetDataType());
38 DType outputDType0 =
ArmNNToDType(outputs[0]->GetDataType());
40 TosaSerializationTensor* inputTensor0 =
new TosaSerializationTensor(input0Name, inputShape0, inputDType0, {});
41 TosaSerializationTensor* inputTensor1 =
new TosaSerializationTensor(input1Name, inputShape1, inputDType1, {});
42 TosaSerializationTensor* outputTensor0 =
new TosaSerializationTensor(outputName, outputShape0, outputDType0, {});
46 return new TosaSerializationBasicBlock(blockName,
48 {inputTensor0, inputTensor1, outputTensor0},
49 {input0Name, input1Name},
std::vector< int32_t > GetTosaTensorShape(const TensorShape &shape)
DType ArmNNToDType(const DataType &type)
std::string GetUniqueTosaMappingID()
TosaSerializationBasicBlock * ConvertAdditionToTosaOperator(const std::vector< const TensorInfo *> &inputs, const std::vector< const TensorInfo *> &outputs, bool isMain)