9 const std::vector<const TensorInfo*>& inputs,
10 const std::vector<const TensorInfo*>& outputs,
13 std::string input0Name = std::string(
"input_");
14 std::string outputName = std::string(
"output0_");
26 TosaSerializationOperator* op =
nullptr;
29 case UnaryOperation::Rsqrt:
31 op =
new TosaSerializationOperator(tosa::Op_RSQRT,
40 throw armnn::Exception(
"ConvertElementwiseUnaryToTosaOperator: Unsupported layer type.");
43 std::vector<TosaSerializationTensor*> tensors;
47 if(input0Name.find(
"input_") != std::string::npos)
50 DType inputDType0 =
ArmNNToDType(inputs[0]->GetDataType());
51 tensors.push_back(
new TosaSerializationTensor(input0Name, inputShape0, inputDType0, {}));
55 DType outputDType0 =
ArmNNToDType(outputs[0]->GetDataType());
57 tensors.push_back(
new TosaSerializationTensor(outputName, outputShape0, outputDType0, {}));
61 return new TosaSerializationBasicBlock(blockName,