/frameworks/ml/nn/runtime/test/specs/V1_0/ |
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_2.mod.py | 19 i2 = Output("op2", "TENSOR_FLOAT32", "{1, 1, 1, 6}") 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, h, w).To(i2) 12 output0 = {i2: # output 0
|
H A D | resize_bilinear_2.mod.py | 20 i2 = Output("op2", "TENSOR_FLOAT32", "{1, 3, 3, 2}") variable 23 model = model.Operation("RESIZE_BILINEAR", i1, h, w).To(i2) 28 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/runtime/test/specs/V1_1/ |
H A D | add_relaxed.mod.py | 20 i2 = Input("op2", "TENSOR_FLOAT32", "{2}") # another vector of 2 float32s variable 23 model = model.Operation("ADD", i1, i2, act).To(i3) 29 i2: # input 1
|