| /frameworks/ml/nn/runtime/test/specs/ |
| H A D | dequantize.mod.py | 4 i2 = Output("op2", "TENSOR_FLOAT32", "{1, 2, 2, 1}") variable
5 model = model.Operation("DEQUANTIZE", i1).To(i2)
11 output0 = {i2: # output 0
|
| H A D | floor.mod.py | 4 i2 = Output("op2", "TENSOR_FLOAT32", "{1, 2, 2, 2}") variable
5 model = model.Operation("FLOOR", i1).To(i2)
12 output0 = {i2: # output 0
|
| H A D | l2_normalization.mod.py | 19 i2 = Output("op2", "TENSOR_FLOAT32", "{1, 1, 1, 3}") variable
22 model = model.Operation("L2_NORMALIZATION", i1).To(i2)
28 output0 = {i2: # output 0
|
| H A D | l2_normalization_large.mod.py | 19 i2 = Output("op2", "TENSOR_FLOAT32", "{1, 2, 2, 3}") variable
22 model = model.Operation("L2_NORMALIZATION", i1).To(i2)
31 output0 = {i2: # output 0
|
| H A D | relu1_float_1.mod.py | 20 i2 = Output("op2", "TENSOR_FLOAT32", "{1, 2, 2, 1}") # output 0 variable
21 model = model.Operation("RELU1", i1).To(i2)
25 output0 = {i2: # output 0
|
| H A D | relu6_float_1.mod.py | 20 i2 = Output("op2", "TENSOR_FLOAT32", "{1, 2, 2, 1}") # output 0 variable
21 model = model.Operation("RELU6", i1).To(i2)
25 output0 = {i2: # output 0
|
| H A D | relu_float_1.mod.py | 20 i2 = Output("op2", "TENSOR_FLOAT32", "{1, 2, 2, 1}") # output 0 variable
21 model = model.Operation("RELU", i1).To(i2)
25 output0 = {i2: # output 0
|
| H A D | tanh.mod.py | 5 i2 = Output("op2", "TENSOR_FLOAT32", "{1, 2, 2, 1}") variable
6 model = model.Operation("TANH", i1).To(i2)
12 output0 = {i2: # output 0
|
| H A D | relu6_quant8_1.mod.py | 20 i2 = Output("op2", "TENSOR_QUANT8_ASYMM", "{1, 2, 2, 1}, 0.5f, 0") # output 0 variable
21 model = model.Operation("RELU6", i1).To(i2)
26 output0 = {i2: # output 0
34 output1 = {i2: # output 0
|
| H A D | add.mod.py | 4 i2 = Input("op2", "TENSOR_FLOAT32", "{2}") # another vector of 2 float32s variable
7 model = model.Operation("ADD", i1, i2, act).To(i3)
12 i2: # input 1
|
| H A D | add_broadcast_quant8.mod.py | 4 i2 = Input("op2", "TENSOR_QUANT8_ASYMM", "{2, 2}, 1.0, 0") variable
7 model = model.Operation("ADD", i1, i2, act).To(i3)
12 i2: # input 1
|
| H A D | add_quant8.mod.py | 4 i2 = Input("op2", "TENSOR_QUANT8_ASYMM", "{2}, 1.0, 0") variable
7 model = model.Operation("ADD", i1, i2, act).To(i3)
12 i2: # input 1
|
| H A D | concat_float_1.mod.py | 20 i2 = Input("op2", "TENSOR_FLOAT32", "{2, 3}") # input tensor 1 variable
23 model = model.Operation("CONCATENATION", i1, i2, axis0).To(r)
27 i2: [7.0, 8.0, 9.0, 10.0, 11.0, 12.0]}
|
| H A D | concat_quant8_1.mod.py | 20 i2 = Input("op2", "TENSOR_QUANT8_ASYMM", "{2, 3}, 0.5f, 0") # input 1 variable
23 model = model.Operation("CONCATENATION", i1, i2, axis1).To(r)
27 i2: [7, 8, 9, 10, 11, 12]}
|
| H A D | mul.mod.py | 4 i2 = Input("op2", "TENSOR_FLOAT32", "{1, 2, 2, 1}") variable
7 model = model.Operation("MUL", i1, i2, act).To(i3)
12 i2: # input 1
|
| H A D | mul_broadcast_quant8.mod.py | 4 i2 = Input("op2", "TENSOR_QUANT8_ASYMM", "{2, 2}, 1.0, 0") variable
7 model = model.Operation("MUL", i1, i2, act).To(i3)
12 i2: # input 1
|
| H A D | mul_quant8.mod.py | 4 i2 = Input("op2", "TENSOR_QUANT8_ASYMM", "{2}, 1.0, 0") variable
7 model = model.Operation("MUL", i1, i2, act).To(i3)
12 i2: # input 1
|
| H A D | mul_relu.mod.py | 4 i2 = Input("op2", "TENSOR_FLOAT32", "{1, 2, 2, 1}") variable
7 model = model.Operation("MUL", i1, i2, act).To(i3)
12 i2: # input 1
|
| H A D | reshape_quant8_weights_as_inputs.mod.py | 5 i2 = Input("op2", "TENSOR_INT32", "{1}") variable
7 model = model.Operation("RESHAPE", i1, i2).To(i3)
14 i2: # input 1
|
| H A D | reshape_weights_as_inputs.mod.py | 4 i2 = Input("op2", "TENSOR_INT32", "{1}") # another vector of 2 float32s variable
6 model = model.Operation("RESHAPE", i1, i2).To(i3)
13 i2: # input 1
|
| H A D | resize_bilinear.mod.py | 4 i2 = Output("op2", "TENSOR_FLOAT32", "{1, 3, 3, 1}") variable
7 model = model.Operation("RESIZE_BILINEAR", i1, w, h).To(i2)
12 output0 = {i2: # output 0
|
| /frameworks/ml/nn/tools/test_generator/tests/P_float/ |
| H A D | addfloat.mod.py | 3 i2 = Input("operand2","TENSOR_FLOAT32", "{3,4}") variable
7 Model().Add(i1, i2).Add(i3).Out(o)
|
| /frameworks/ml/nn/tools/test_generator/tests/P_full/ |
| H A D | addfloat.mod.py | 4 i2 = Input("op2", "TENSOR_FLOAT32", "{2}") # another vector of 2 float32s variable
7 model = model.Operation("ADD", i1, i2, b0).To(i3)
12 i2: # input 1
|
| /frameworks/ml/nn/tools/test_generator/tests/P_weird/ |
| H A D | weird_add.mod.py | 4 i2 = Input("op2", "TENSOR_FLOAT32", "{2}") # another vector of 2 float32s variable
10 model = model.Operation("ADD", i1, i2, b0).To(tmp)
11 model = model.Operation("ADD", tmp, i2, b0).To(tmp2)
17 i2: # input 1
|
| /frameworks/ml/nn/runtime/test/generated/models/ |
| H A D | conv_1_h3_w2_SAME.model.py | 6 i2 = Input("op2", "TENSOR_FLOAT32", "{1, 8, 8, 3}") # input 0 variable
10 model = model.Conv(i2, i0, i1, i4, i5, i6, i7).To(i3)
|