externals/nnapi_test_generator/tests/P_vts_operands/addfloat.mod.py


1
2
3
4
5
6
7
8
9
10
11
12

# Force VTS mode
Configuration.vts = True
# model
i1 = Input("operand1","TENSOR_FLOAT32", "{3,4}")
i2 = Input("operand2","TENSOR_FLOAT32", "{3,4}")
i3 = Input("operand3","TENSOR_FLOAT32", "{3,4}")
Parameter("p1", "TENSOR_QUANT8_ASYMM", "{1, 2, 3}", [1, 2, 3, 4, 5, 6])
Parameter("p2", "TENSOR_FLOAT32", "{}", [42.0])
o = Output("operand4","TENSOR_FLOAT32", "{3,4}")

Model().Add(i1, i2).Add(i3).Out(o)