/frameworks/ml/nn/runtime/test/specs/V1_3/ |
D | pad_quant8_signed.mod.py | 19 input0 = Input("input0", "TENSOR_FLOAT32", "{1, 1, 2, 3}") variable 26 model = Model().Operation("PAD", input0, paddings).To(output0) 29 input0: ("TENSOR_QUANT8_ASYMM_SIGNED", 2.3, -128), 34 input0: [1.0, 2.0, 3.0, 47 input0 = Input("input0", "TENSOR_QUANT8_ASYMM_SIGNED", "{3}, 2.3, -128") variable 51 model = Model().Operation("PAD", input0, paddings).To(output0) 54 input0: [-127, -126, -125], 60 input0 = Input("input0", "TENSOR_QUANT8_ASYMM_SIGNED", "{1, 2, 3, 1}, 2.3, -128") variable 67 model = Model().Operation("PAD", input0, paddings).To(output0) 70 input0: [-127, -126, -125, [all …]
|
D | gather_quant8_signed.mod.py | 17 input0 = Input("input0", "TENSOR_FLOAT32", "{1, 3, 2}") variable 22 model = Model().Operation("GATHER", input0, axis, indices).To(output0) 25 input0: ["TENSOR_QUANT8_ASYMM_SIGNED", 0.5, -1], 30 input0: [1, 2, 46 def test(input0, axis, indices, output0, input_data, output_data): argument 47 model = Model().Operation("GATHER", input0, axis, indices).To(output0) 50 input0: ["TENSOR_QUANT8_ASYMM_SIGNED", 0.5, -1], 55 input0: input_data, 60 input0=Input("input0", "TENSOR_FLOAT32", "{2, 2}"), 71 input0=Input("input0", "TENSOR_FLOAT32", "{2, 2}"), [all …]
|
D | argmax_quant8_signed.mod.py | 17 input0 = Input("input0", "TENSOR_FLOAT32", "{2, 2}") variable 21 model = Model().Operation("ARGMAX", input0, axis).To(output0) 24 input0: ["TENSOR_QUANT8_ASYMM_SIGNED", 1.0, 0], 28 input0: [1.0, 2.0, 35 input0 = Input("input0", "TENSOR_FLOAT32", "{2, 2}") variable 39 model = Model().Operation("ARGMAX", input0, axis).To(output0) 42 input0: ["TENSOR_QUANT8_ASYMM_SIGNED", 1.0, 0], 46 input0: [1.0, 2.0, 55 input0 = Input("input0", "TENSOR_FLOAT32", "{2, 2}") variable 59 model = Model().Operation("ARGMAX", input0, axis).To(output0) [all …]
|
D | argmin_quant8_signed.mod.py | 18 input0 = Input("input0", "TENSOR_FLOAT32", "{2, 2}") variable 22 model = Model().Operation("ARGMIN", input0, axis).To(output0) 25 input0: ["TENSOR_QUANT8_ASYMM_SIGNED", 1.0, 0], 29 input0: [1.0, 2.0, 36 input0 = Input("input0", "TENSOR_FLOAT32", "{2, 2}") variable 40 model = Model().Operation("ARGMIN", input0, axis).To(output0) 43 input0: ["TENSOR_QUANT8_ASYMM_SIGNED", 1.0, 0], 47 input0: [1.0, 2.0, 54 input0 = Input("input0", "TENSOR_FLOAT32", "{2, 2}") variable 58 model = Model().Operation("ARGMIN", input0, axis).To(output0) [all …]
|
D | tile_quant8_signed.mod.py | 17 input0 = Input("input0", "TENSOR_FLOAT32", "{3}") variable 21 model = Model().Operation("TILE", input0, multipliers).To(output0) 28 input0: ["TENSOR_QUANT8_ASYMM_SIGNED", 0.5, -1], 33 input0: input_values, 40 input0 = Input("input0", "TENSOR_FLOAT32", "{2, 3}") variable 44 model = Model().Operation("TILE", input0, multipliers).To(output0) 55 input0: ["TENSOR_QUANT8_ASYMM_SIGNED", 0.5, -1], 60 input0: input_values, 67 input0 = Input("input0", "TENSOR_FLOAT32", "{1, 2, 3}") variable 71 model = Model().Operation("TILE", input0, multipliers).To(output0) [all …]
|
D | split_quant8_signed.mod.py | 17 input0 = Input("input0", "TENSOR_QUANT8_ASYMM_SIGNED", "{6}, 1.0, -128") variable 24 model = Model().Operation("SPLIT", input0, axis, num_splits).To( 28 input_dict = {input0: [-127, -126, -125, -124, -123, -122]} 40 input0 = Input("input0", "TENSOR_QUANT8_ASYMM_SIGNED", "{2, 3}, 2.0, -125") variable 46 model = Model().Operation("SPLIT", input0, axis, num_splits).To( 50 input_dict = {input0: [-127, -126, -125, -124, -123, -122]} 61 input0 = Input("input0", "TENSOR_QUANT8_ASYMM_SIGNED", "{2, 3}, 2.0, -125") variable 68 model = Model().Operation("SPLIT", input0, axis, num_splits).To( 72 input_dict = {input0: [-127, -126, -125, -124, -123, -122]} 84 input0 = Input("input0", "TENSOR_QUANT8_ASYMM_SIGNED", "{2, 2, 2}, 1.0, -128") variable [all …]
|
D | strided_slice_quant8_signed.mod.py | 31 input0 = {i1: # input 0 variable 38 Example((input0, output0)) 56 input0 = {i1: # input 0 variable 63 Example((input0, output0)) 81 input0 = {i1: # input 0 variable 88 Example((input0, output0)) 106 input0 = {i1: # input 0 variable 113 Example((input0, output0)) 131 input0 = {i1: # input 0 variable 138 Example((input0, output0)) [all …]
|
D | hard_swish.mod.py | 18 def test(name, input0, output0, input0_data, output0_data): argument 19 model = Model().Operation("HARD_SWISH", input0).To(output0) 21 input0: ["TENSOR_QUANT8_ASYMM", 0.078125, 128], 25 input0: ["TENSOR_QUANT8_ASYMM_SIGNED", 0.078125, 0], 29 input0: ["TENSOR_QUANT8_ASYMM", 0.078125, 128], 33 input0: ["TENSOR_QUANT8_ASYMM_SIGNED", 0.078125, 0], 37 input0: input0_data, 48 input0=Input("input0", "TENSOR_FLOAT32", "{40}"), 69 input0=Input("input0", "TENSOR_FLOAT32", "{1, 2, 2, 2, 5}"),
|
D | minimum_quant8_signed.mod.py | 17 def test(name, input0, input1, output0, input0_data, input1_data, output_data): argument 18 model = Model().Operation("MINIMUM", input0, input1).To(output0) 21 input0: ["TENSOR_QUANT8_ASYMM_SIGNED", 0.5, -1], 27 input0: input0_data, 35 input0=Input("input0", "TENSOR_FLOAT32", "{3, 1, 2}"), 45 input0=Input("input0", "TENSOR_FLOAT32", "{3, 1, 2}"), 54 input0 = Input("input0", "TENSOR_QUANT8_ASYMM_SIGNED", "{2}, 1.0f, 0") variable 57 model = Model().Operation("MINIMUM", input0, input1).To(output0) 60 input0: [-68, 0],
|
D | maximum_quant8_signed.mod.py | 17 def test(name, input0, input1, output0, input0_data, input1_data, output_data): argument 18 model = Model().Operation("MAXIMUM", input0, input1).To(output0) 21 input0: ["TENSOR_QUANT8_ASYMM_SIGNED", 0.5, -1], 27 input0: input0_data, 35 input0=Input("input0", "TENSOR_FLOAT32", "{3, 1, 2}"), 45 input0=Input("input0", "TENSOR_FLOAT32", "{3, 1, 2}"), 55 input0 = Input("input0", "TENSOR_QUANT8_ASYMM_SIGNED", "{2}, 1.0f, 0") variable 58 model = Model().Operation("MAXIMUM", input0, input1).To(output0) 61 input0: [-68, 0],
|
D | resize_nearest_neighbor_v1_3.mod.py | 16 def test(name, input0, output_width, output_height, layout, align_corners, argument 18 model = Model().Operation("RESIZE_NEAREST_NEIGHBOR", input0, output_width, 22 input0: ["TENSOR_QUANT8_ASYMM", 0.5, 128], 26 input0: ["TENSOR_QUANT8_ASYMM_SIGNED", 0.5, 0], 30 input0: input0_data, 39 input0=Input("input0", "TENSOR_FLOAT32", "{1, 2, 5, 1}"), 52 input0=Input("input0", "TENSOR_FLOAT32", "{1, 2, 2, 1}"), 65 input0=Input("input0", "TENSOR_FLOAT32", "{1, 2, 2, 1}"), 78 input0=Input("input0", "TENSOR_FLOAT32", "{1, 2, 2, 1}"), 91 input0=Input("input0", "TENSOR_FLOAT32", "{1, 4, 4, 1}"), [all …]
|
D | reduce_max_quant8_signed.mod.py | 17 def test(input0, output0, axes, keep_dims, input_data, output_data): argument 18 model = Model().Operation("REDUCE_MAX", input0, axes, keep_dims).To(output0) 20 input0: ["TENSOR_QUANT8_ASYMM_SIGNED", 0.5, -1], 24 input0: input_data, 29 input0=Input("input0", "TENSOR_FLOAT32", "{3, 2}"), 42 input0=Input("input0", "TENSOR_FLOAT32", "{1}"), 51 input0=Input("input0", "TENSOR_FLOAT32", "{4, 3, 2}"), 62 input0=Input("input0", "TENSOR_FLOAT32", "{4, 3, 2}"),
|
D | reduce_min_quant8_signed.mod.py | 17 def test(input0, output0, axes, keep_dims, input_data, output_data): argument 18 model = Model().Operation("REDUCE_MIN", input0, axes, keep_dims).To(output0) 20 input0: ["TENSOR_QUANT8_ASYMM_SIGNED", 0.5, -1], 24 input0: input_data, 29 input0=Input("input0", "TENSOR_FLOAT32", "{3, 2}"), 42 input0=Input("input0", "TENSOR_FLOAT32", "{1}"), 51 input0=Input("input0", "TENSOR_FLOAT32", "{4, 3, 2}"), 62 input0=Input("input0", "TENSOR_FLOAT32", "{4, 3, 2}"),
|
/frameworks/ml/nn/runtime/test/specs/V1_2/ |
D | gather.mod.py | 17 def test(input0, axis, indices, output0, input_data, output_data): argument 18 model = Model().Operation("GATHER", input0, axis, indices).To(output0) 21 input0: ["TENSOR_QUANT8_ASYMM", 0.5, 127], 26 input0: ["TENSOR_INT32"], 31 input0: ["TENSOR_FLOAT16"], 36 input0: input_data, 41 input0=Input("input0", "TENSOR_FLOAT32", "{2, 2}"), 52 input0=Input("input0", "TENSOR_FLOAT32", "{2, 2}"), 62 input0=Input("input0", "TENSOR_FLOAT32", "{3}"), 71 input0=Input("input0", "TENSOR_FLOAT32", "{3}"), [all …]
|
D | maximum.mod.py | 17 def test(name, input0, input1, output0, input0_data, input1_data, output_data): argument 18 model = Model().Operation("MAXIMUM", input0, input1).To(output0) 21 input0: ["TENSOR_QUANT8_ASYMM", 0.5, 127], 27 input0: input0_data, 35 input0=Input("input0", "TENSOR_FLOAT32", "{3, 1, 2}"), 45 input0=Input("input0", "TENSOR_FLOAT32", "{3, 1, 2}"), 55 input0 = Input("input0", "TENSOR_QUANT8_ASYMM", "{2}, 1.0f, 128") variable 58 model = Model().Operation("MAXIMUM", input0, input1).To(output0) 61 input0: [60, 128],
|
D | minimum.mod.py | 17 def test(name, input0, input1, output0, input0_data, input1_data, output_data): argument 18 model = Model().Operation("MINIMUM", input0, input1).To(output0) 21 input0: ["TENSOR_QUANT8_ASYMM", 0.5, 127], 27 input0: input0_data, 35 input0=Input("input0", "TENSOR_FLOAT32", "{3, 1, 2}"), 45 input0=Input("input0", "TENSOR_FLOAT32", "{3, 1, 2}"), 55 input0 = Input("input0", "TENSOR_QUANT8_ASYMM", "{2}, 1.0f, 128") variable 58 model = Model().Operation("MINIMUM", input0, input1).To(output0) 61 input0: [60, 128],
|
D | dequantize_v1_2.mod.py | 18 def test(name, input0, output0, input0_data, output0_data): argument 19 model = Model().Operation("DEQUANTIZE", input0).To(output0) 21 input0: input0_data, 30 input0=Input("input0", "TENSOR_QUANT8_ASYMM", "{10}, 0.5, 127"), 38 input0=Input("input0", "TENSOR_QUANT8_ASYMM", "{2, 5}, 0.5, 127"), 46 input0=Input("input0", "TENSOR_QUANT8_SYMM", "{2, 2, 2}, 0.5, 0"), 54 input0=Input("input0", "TENSOR_QUANT8_SYMM", "{2, 1, 2, 2}, 0.5, 0"), 62 input0=Input( 78 input0=Input( 108 input0 = {i1: # input 0 variable [all …]
|
D | expand_dims.mod.py | 17 input0 = Input("input0", "TENSOR_FLOAT32", "{2, 2}") variable 24 model0 = Model().Operation("EXPAND_DIMS", input0, 0).To(output0) 25 model1 = Model().Operation("EXPAND_DIMS", input0, 1).To(output1) 26 model2 = Model().Operation("EXPAND_DIMS", input0, 2).To(output2) 27 model3 = Model().Operation("EXPAND_DIMS", input0, -1).To(output3) 36 input0: ["TENSOR_QUANT8_ASYMM", 0.5, 127], 41 input0: ["TENSOR_INT32"], 46 input0: data,
|
D | reduce_max.mod.py | 17 def test(input0, output0, axes, keep_dims, input_data, output_data): argument 18 model = Model().Operation("REDUCE_MAX", input0, axes, keep_dims).To(output0) 20 input0: ["TENSOR_QUANT8_ASYMM", 0.5, 127], 24 input0: input_data, 29 input0=Input("input0", "TENSOR_FLOAT32", "{3, 2}"), 42 input0=Input("input0", "TENSOR_FLOAT32", "{1}"), 51 input0=Input("input0", "TENSOR_FLOAT32", "{4, 3, 2}"), 62 input0=Input("input0", "TENSOR_FLOAT32", "{4, 3, 2}"),
|
D | reduce_min.mod.py | 17 def test(input0, output0, axes, keep_dims, input_data, output_data): argument 18 model = Model().Operation("REDUCE_MIN", input0, axes, keep_dims).To(output0) 20 input0: ["TENSOR_QUANT8_ASYMM", 0.5, 127], 24 input0: input_data, 29 input0=Input("input0", "TENSOR_FLOAT32", "{3, 2}"), 42 input0=Input("input0", "TENSOR_FLOAT32", "{1}"), 51 input0=Input("input0", "TENSOR_FLOAT32", "{4, 3, 2}"), 62 input0=Input("input0", "TENSOR_FLOAT32", "{4, 3, 2}"),
|
D | equal.mod.py | 16 def test(name, input0, input1, output0, input0_data, input1_data, output_data, do_variations=True): argument 17 model = Model().Operation("EQUAL", input0, input1).To(output0) 19 input0: input0_data, 28 input0=Input("input0", "TENSOR_FLOAT32", "{3}"), 38 input0=Input("input0", "TENSOR_FLOAT32", "{2, 1}"), 48 input0=Input("input0", ("TENSOR_QUANT8_ASYMM", [3], 1.0, 128)), 59 input0=Input("input0", ("TENSOR_QUANT8_ASYMM", [3], 1.0, 128)), 70 input0=Input("input0", ("TENSOR_QUANT8_ASYMM", [1], 1.64771, 31)), 81 input0=Input("input0", ("TENSOR_QUANT8_ASYMM", [1], 1.49725, 240)), 92 input0=Input("input0", "TENSOR_BOOL8", "{4}"),
|
D | less_equal.mod.py | 16 def test(name, input0, input1, output0, input0_data, input1_data, output_data, do_variations=True): argument 17 model = Model().Operation("LESS_EQUAL", input0, input1).To(output0) 19 input0: input0_data, 28 input0=Input("input0", "TENSOR_FLOAT32", "{3}"), 38 input0=Input("input0", "TENSOR_FLOAT32", "{2, 1}"), 48 input0=Input("input0", ("TENSOR_QUANT8_ASYMM", [3], 1.0, 128)), 59 input0=Input("input0", ("TENSOR_QUANT8_ASYMM", [3], 1.0, 128)), 70 input0=Input("input0", ("TENSOR_QUANT8_ASYMM", [1], 1.64771, 31)), 81 input0=Input("input0", ("TENSOR_QUANT8_ASYMM", [1], 1.49725, 240)), 92 input0=Input("input0", "TENSOR_BOOL8", "{4}"),
|
D | greater.mod.py | 16 def test(name, input0, input1, output0, input0_data, input1_data, output_data, do_variations=True): argument 17 model = Model().Operation("GREATER", input0, input1).To(output0) 19 input0: input0_data, 28 input0=Input("input0", "TENSOR_FLOAT32", "{3}"), 38 input0=Input("input0", "TENSOR_FLOAT32", "{2, 1}"), 48 input0=Input("input0", ("TENSOR_QUANT8_ASYMM", [3], 1.0, 128)), 59 input0=Input("input0", ("TENSOR_QUANT8_ASYMM", [3], 1.0, 128)), 70 input0=Input("input0", ("TENSOR_QUANT8_ASYMM", [1], 1.64771, 31)), 81 input0=Input("input0", ("TENSOR_QUANT8_ASYMM", [1], 1.49725, 240)), 92 input0=Input("input0", "TENSOR_BOOL8", "{4}"),
|
D | not_equal.mod.py | 16 def test(name, input0, input1, output0, input0_data, input1_data, output_data, do_variations=True): argument 17 model = Model().Operation("NOT_EQUAL", input0, input1).To(output0) 19 input0: input0_data, 28 input0=Input("input0", "TENSOR_FLOAT32", "{3}"), 38 input0=Input("input0", "TENSOR_FLOAT32", "{2, 1}"), 48 input0=Input("input0", ("TENSOR_QUANT8_ASYMM", [3], 1.0, 128)), 59 input0=Input("input0", ("TENSOR_QUANT8_ASYMM", [3], 1.0, 128)), 70 input0=Input("input0", ("TENSOR_QUANT8_ASYMM", [1], 1.64771, 31)), 81 input0=Input("input0", ("TENSOR_QUANT8_ASYMM", [1], 1.49725, 240)), 92 input0=Input("input0", "TENSOR_BOOL8", "{4}"),
|
D | less.mod.py | 16 def test(name, input0, input1, output0, input0_data, input1_data, output_data, do_variations=True): argument 17 model = Model().Operation("LESS", input0, input1).To(output0) 19 input0: input0_data, 28 input0=Input("input0", "TENSOR_FLOAT32", "{3}"), 38 input0=Input("input0", "TENSOR_FLOAT32", "{2, 1}"), 48 input0=Input("input0", ("TENSOR_QUANT8_ASYMM", [3], 1.0, 128)), 59 input0=Input("input0", ("TENSOR_QUANT8_ASYMM", [3], 1.0, 128)), 70 input0=Input("input0", ("TENSOR_QUANT8_ASYMM", [1], 1.64771, 31)), 81 input0=Input("input0", ("TENSOR_QUANT8_ASYMM", [1], 1.49725, 240)), 92 input0=Input("input0", "TENSOR_BOOL8", "{4}"),
|