| /frameworks/ml/nn/runtime/test/generated/models/ |
| D | expand_dims.model.cpp | 8 auto input0 = model->addOperand(&type0); in CreateModel() local
14 model->addOperation(ANEURALNETWORKS_EXPAND_DIMS, {input0, param}, {output}); in CreateModel()
17 {input0}, in CreateModel()
32 auto input0 = model->addOperand(&type0); in CreateModel_relaxed() local
38 model->addOperation(ANEURALNETWORKS_EXPAND_DIMS, {input0, param}, {output}); in CreateModel_relaxed()
41 {input0}, in CreateModel_relaxed()
58 auto input0 = model->addOperand(&type5); in CreateModel_quant8() local
64 model->addOperation(ANEURALNETWORKS_EXPAND_DIMS, {input0, param}, {output}); in CreateModel_quant8()
67 {input0}, in CreateModel_quant8()
82 auto input0 = model->addOperand(&type7); in CreateModel_int32() local
[all …]
|
| D | neg.model.cpp | 6 auto input0 = model->addOperand(&type0); in CreateModel() local
9 model->addOperation(ANEURALNETWORKS_NEG, {input0}, {output0}); in CreateModel()
12 {input0}, in CreateModel()
25 auto input0 = model->addOperand(&type0); in CreateModel_relaxed() local
28 model->addOperation(ANEURALNETWORKS_NEG, {input0}, {output0}); in CreateModel_relaxed()
31 {input0}, in CreateModel_relaxed()
46 auto input0 = model->addOperand(&type1); in CreateModel_float16() local
49 model->addOperation(ANEURALNETWORKS_NEG, {input0}, {output0}); in CreateModel_float16()
52 {input0}, in CreateModel_float16()
65 auto input0 = model->addOperand(&type2); in CreateModel_int32() local
[all …]
|
| D | tile_2.model.cpp | 8 auto input0 = model->addOperand(&type0); in CreateModel() local
12 model->addOperation(ANEURALNETWORKS_TILE, {input0, multipliers}, {output0}); in CreateModel()
15 {input0, multipliers}, in CreateModel()
30 auto input0 = model->addOperand(&type0); in CreateModel_relaxed() local
34 model->addOperation(ANEURALNETWORKS_TILE, {input0, multipliers}, {output0}); in CreateModel_relaxed()
37 {input0, multipliers}, in CreateModel_relaxed()
54 auto input0 = model->addOperand(&type3); in CreateModel_float16() local
58 model->addOperation(ANEURALNETWORKS_TILE, {input0, multipliers}, {output0}); in CreateModel_float16()
61 {input0, multipliers}, in CreateModel_float16()
76 auto input0 = model->addOperand(&type5); in CreateModel_quant8() local
[all …]
|
| D | tile_3.model.cpp | 8 auto input0 = model->addOperand(&type0); in CreateModel() local
12 model->addOperation(ANEURALNETWORKS_TILE, {input0, multipliers}, {output0}); in CreateModel()
15 {input0, multipliers}, in CreateModel()
30 auto input0 = model->addOperand(&type0); in CreateModel_relaxed() local
34 model->addOperation(ANEURALNETWORKS_TILE, {input0, multipliers}, {output0}); in CreateModel_relaxed()
37 {input0, multipliers}, in CreateModel_relaxed()
54 auto input0 = model->addOperand(&type3); in CreateModel_float16() local
58 model->addOperation(ANEURALNETWORKS_TILE, {input0, multipliers}, {output0}); in CreateModel_float16()
61 {input0, multipliers}, in CreateModel_float16()
76 auto input0 = model->addOperand(&type5); in CreateModel_quant8() local
[all …]
|
| D | sqrt.model.cpp | 6 auto input0 = model->addOperand(&type0); in CreateModel() local
9 model->addOperation(ANEURALNETWORKS_SQRT, {input0}, {output0}); in CreateModel()
12 {input0}, in CreateModel()
25 auto input0 = model->addOperand(&type0); in CreateModel_relaxed() local
28 model->addOperation(ANEURALNETWORKS_SQRT, {input0}, {output0}); in CreateModel_relaxed()
31 {input0}, in CreateModel_relaxed()
46 auto input0 = model->addOperand(&type1); in CreateModel_float16() local
49 model->addOperation(ANEURALNETWORKS_SQRT, {input0}, {output0}); in CreateModel_float16()
52 {input0}, in CreateModel_float16()
66 auto input0 = model->addOperand(&type0); in CreateModel_dynamic_output_shape() local
[all …]
|
| D | abs.model.cpp | 6 auto input0 = model->addOperand(&type0); in CreateModel() local
9 model->addOperation(ANEURALNETWORKS_ABS, {input0}, {output0}); in CreateModel()
12 {input0}, in CreateModel()
25 auto input0 = model->addOperand(&type0); in CreateModel_relaxed() local
28 model->addOperation(ANEURALNETWORKS_ABS, {input0}, {output0}); in CreateModel_relaxed()
31 {input0}, in CreateModel_relaxed()
46 auto input0 = model->addOperand(&type1); in CreateModel_float16() local
49 model->addOperation(ANEURALNETWORKS_ABS, {input0}, {output0}); in CreateModel_float16()
52 {input0}, in CreateModel_float16()
66 auto input0 = model->addOperand(&type0); in CreateModel_dynamic_output_shape() local
[all …]
|
| D | exp.model.cpp | 6 auto input0 = model->addOperand(&type0); in CreateModel() local
9 model->addOperation(ANEURALNETWORKS_EXP, {input0}, {output0}); in CreateModel()
12 {input0}, in CreateModel()
25 auto input0 = model->addOperand(&type0); in CreateModel_relaxed() local
28 model->addOperation(ANEURALNETWORKS_EXP, {input0}, {output0}); in CreateModel_relaxed()
31 {input0}, in CreateModel_relaxed()
46 auto input0 = model->addOperand(&type1); in CreateModel_float16() local
49 model->addOperation(ANEURALNETWORKS_EXP, {input0}, {output0}); in CreateModel_float16()
52 {input0}, in CreateModel_float16()
66 auto input0 = model->addOperand(&type0); in CreateModel_dynamic_output_shape() local
[all …]
|
| D | rsqrt.model.cpp | 6 auto input0 = model->addOperand(&type0); in CreateModel() local
9 model->addOperation(ANEURALNETWORKS_RSQRT, {input0}, {output0}); in CreateModel()
12 {input0}, in CreateModel()
25 auto input0 = model->addOperand(&type0); in CreateModel_relaxed() local
28 model->addOperation(ANEURALNETWORKS_RSQRT, {input0}, {output0}); in CreateModel_relaxed()
31 {input0}, in CreateModel_relaxed()
46 auto input0 = model->addOperand(&type1); in CreateModel_float16() local
49 model->addOperation(ANEURALNETWORKS_RSQRT, {input0}, {output0}); in CreateModel_float16()
52 {input0}, in CreateModel_float16()
66 auto input0 = model->addOperand(&type0); in CreateModel_dynamic_output_shape() local
[all …]
|
| D | log.model.cpp | 6 auto input0 = model->addOperand(&type0); in CreateModel() local
9 model->addOperation(ANEURALNETWORKS_LOG, {input0}, {output0}); in CreateModel()
12 {input0}, in CreateModel()
25 auto input0 = model->addOperand(&type0); in CreateModel_relaxed() local
28 model->addOperation(ANEURALNETWORKS_LOG, {input0}, {output0}); in CreateModel_relaxed()
31 {input0}, in CreateModel_relaxed()
46 auto input0 = model->addOperand(&type1); in CreateModel_float16() local
49 model->addOperation(ANEURALNETWORKS_LOG, {input0}, {output0}); in CreateModel_float16()
52 {input0}, in CreateModel_float16()
66 auto input0 = model->addOperand(&type0); in CreateModel_dynamic_output_shape() local
[all …]
|
| D | sin.model.cpp | 6 auto input0 = model->addOperand(&type0); in CreateModel() local
9 model->addOperation(ANEURALNETWORKS_SIN, {input0}, {output0}); in CreateModel()
12 {input0}, in CreateModel()
25 auto input0 = model->addOperand(&type0); in CreateModel_relaxed() local
28 model->addOperation(ANEURALNETWORKS_SIN, {input0}, {output0}); in CreateModel_relaxed()
31 {input0}, in CreateModel_relaxed()
46 auto input0 = model->addOperand(&type1); in CreateModel_float16() local
49 model->addOperation(ANEURALNETWORKS_SIN, {input0}, {output0}); in CreateModel_float16()
52 {input0}, in CreateModel_float16()
66 auto input0 = model->addOperand(&type0); in CreateModel_dynamic_output_shape() local
[all …]
|
| D | argmin_1.model.cpp | 8 auto input0 = model->addOperand(&type0); in CreateModel() local
14 model->addOperation(ANEURALNETWORKS_ARGMIN, {input0, axis}, {output}); in CreateModel()
17 {input0}, in CreateModel()
32 auto input0 = model->addOperand(&type0); in CreateModel_relaxed() local
38 model->addOperation(ANEURALNETWORKS_ARGMIN, {input0, axis}, {output}); in CreateModel_relaxed()
41 {input0}, in CreateModel_relaxed()
58 auto input0 = model->addOperand(&type3); in CreateModel_float16() local
64 model->addOperation(ANEURALNETWORKS_ARGMIN, {input0, axis}, {output}); in CreateModel_float16()
67 {input0}, in CreateModel_float16()
82 auto input0 = model->addOperand(&type4); in CreateModel_int32() local
[all …]
|
| D | argmax_3.model.cpp | 8 auto input0 = model->addOperand(&type0); in CreateModel() local
14 model->addOperation(ANEURALNETWORKS_ARGMAX, {input0, axis}, {output}); in CreateModel()
17 {input0}, in CreateModel()
32 auto input0 = model->addOperand(&type0); in CreateModel_relaxed() local
38 model->addOperation(ANEURALNETWORKS_ARGMAX, {input0, axis}, {output}); in CreateModel_relaxed()
41 {input0}, in CreateModel_relaxed()
58 auto input0 = model->addOperand(&type3); in CreateModel_float16() local
64 model->addOperation(ANEURALNETWORKS_ARGMAX, {input0, axis}, {output}); in CreateModel_float16()
67 {input0}, in CreateModel_float16()
82 auto input0 = model->addOperand(&type4); in CreateModel_int32() local
[all …]
|
| D | argmin_3.model.cpp | 8 auto input0 = model->addOperand(&type0); in CreateModel() local
14 model->addOperation(ANEURALNETWORKS_ARGMIN, {input0, axis}, {output}); in CreateModel()
17 {input0}, in CreateModel()
32 auto input0 = model->addOperand(&type0); in CreateModel_relaxed() local
38 model->addOperation(ANEURALNETWORKS_ARGMIN, {input0, axis}, {output}); in CreateModel_relaxed()
41 {input0}, in CreateModel_relaxed()
58 auto input0 = model->addOperand(&type3); in CreateModel_float16() local
64 model->addOperation(ANEURALNETWORKS_ARGMIN, {input0, axis}, {output}); in CreateModel_float16()
67 {input0}, in CreateModel_float16()
82 auto input0 = model->addOperand(&type4); in CreateModel_int32() local
[all …]
|
| D | argmax_1.model.cpp | 8 auto input0 = model->addOperand(&type0); in CreateModel() local
14 model->addOperation(ANEURALNETWORKS_ARGMAX, {input0, axis}, {output}); in CreateModel()
17 {input0}, in CreateModel()
32 auto input0 = model->addOperand(&type0); in CreateModel_relaxed() local
38 model->addOperation(ANEURALNETWORKS_ARGMAX, {input0, axis}, {output}); in CreateModel_relaxed()
41 {input0}, in CreateModel_relaxed()
58 auto input0 = model->addOperand(&type3); in CreateModel_float16() local
64 model->addOperation(ANEURALNETWORKS_ARGMAX, {input0, axis}, {output}); in CreateModel_float16()
67 {input0}, in CreateModel_float16()
82 auto input0 = model->addOperand(&type4); in CreateModel_int32() local
[all …]
|
| D | argmin_2.model.cpp | 8 auto input0 = model->addOperand(&type0); in CreateModel() local
14 model->addOperation(ANEURALNETWORKS_ARGMIN, {input0, axis}, {output}); in CreateModel()
17 {input0}, in CreateModel()
32 auto input0 = model->addOperand(&type0); in CreateModel_relaxed() local
38 model->addOperation(ANEURALNETWORKS_ARGMIN, {input0, axis}, {output}); in CreateModel_relaxed()
41 {input0}, in CreateModel_relaxed()
58 auto input0 = model->addOperand(&type3); in CreateModel_float16() local
64 model->addOperation(ANEURALNETWORKS_ARGMIN, {input0, axis}, {output}); in CreateModel_float16()
67 {input0}, in CreateModel_float16()
82 auto input0 = model->addOperand(&type4); in CreateModel_int32() local
[all …]
|
| D | argmax_2.model.cpp | 8 auto input0 = model->addOperand(&type0); in CreateModel() local
14 model->addOperation(ANEURALNETWORKS_ARGMAX, {input0, axis}, {output}); in CreateModel()
17 {input0}, in CreateModel()
32 auto input0 = model->addOperand(&type0); in CreateModel_relaxed() local
38 model->addOperation(ANEURALNETWORKS_ARGMAX, {input0, axis}, {output}); in CreateModel_relaxed()
41 {input0}, in CreateModel_relaxed()
58 auto input0 = model->addOperand(&type3); in CreateModel_float16() local
64 model->addOperation(ANEURALNETWORKS_ARGMAX, {input0, axis}, {output}); in CreateModel_float16()
67 {input0}, in CreateModel_float16()
82 auto input0 = model->addOperand(&type4); in CreateModel_int32() local
[all …]
|
| D | tile_1.model.cpp | 8 auto input0 = model->addOperand(&type0); in CreateModel() local
12 model->addOperation(ANEURALNETWORKS_TILE, {input0, multipliers}, {output0}); in CreateModel()
15 {input0, multipliers}, in CreateModel()
30 auto input0 = model->addOperand(&type0); in CreateModel_relaxed() local
34 model->addOperation(ANEURALNETWORKS_TILE, {input0, multipliers}, {output0}); in CreateModel_relaxed()
37 {input0, multipliers}, in CreateModel_relaxed()
54 auto input0 = model->addOperand(&type3); in CreateModel_float16() local
58 model->addOperation(ANEURALNETWORKS_TILE, {input0, multipliers}, {output0}); in CreateModel_float16()
61 {input0, multipliers}, in CreateModel_float16()
76 auto input0 = model->addOperand(&type5); in CreateModel_quant8() local
[all …]
|
| D | sub_v1_2_broadcast.model.cpp | 8 auto input0 = model->addOperand(&type0); in CreateModel_none() local
15 model->addOperation(ANEURALNETWORKS_SUB, {input0, input1, act}, {output0}); in CreateModel_none()
18 {input0, input1}, in CreateModel_none()
33 auto input0 = model->addOperand(&type0); in CreateModel_relu() local
40 model->addOperation(ANEURALNETWORKS_SUB, {input0, input1, act}, {output0}); in CreateModel_relu()
43 {input0, input1}, in CreateModel_relu()
58 auto input0 = model->addOperand(&type0); in CreateModel_relu1() local
65 model->addOperation(ANEURALNETWORKS_SUB, {input0, input1, act}, {output0}); in CreateModel_relu1()
68 {input0, input1}, in CreateModel_relu1()
83 auto input0 = model->addOperand(&type0); in CreateModel_relu6() local
[all …]
|
| D | pad_all_dims.model.cpp | 8 auto input0 = model->addOperand(&type0); in CreateModel() local
14 model->addOperation(ANEURALNETWORKS_PAD, {input0, paddings}, {output0}); in CreateModel()
17 {input0}, in CreateModel()
32 auto input0 = model->addOperand(&type3); in CreateModel_float16() local
38 model->addOperation(ANEURALNETWORKS_PAD, {input0, paddings}, {output0}); in CreateModel_float16()
41 {input0}, in CreateModel_float16()
56 auto input0 = model->addOperand(&type0); in CreateModel_relaxed() local
62 model->addOperation(ANEURALNETWORKS_PAD, {input0, paddings}, {output0}); in CreateModel_relaxed()
65 {input0}, in CreateModel_relaxed()
82 auto input0 = model->addOperand(&type5); in CreateModel_quant8() local
[all …]
|
| D | gather_higher_rank.model.cpp | 9 auto input0 = model->addOperand(&type0); in CreateModel() local
16 model->addOperation(ANEURALNETWORKS_GATHER, {input0, param, indices}, {output0}); in CreateModel()
19 {input0, indices}, in CreateModel()
35 auto input0 = model->addOperand(&type0); in CreateModel_relaxed() local
42 model->addOperation(ANEURALNETWORKS_GATHER, {input0, param, indices}, {output0}); in CreateModel_relaxed()
45 {input0, indices}, in CreateModel_relaxed()
63 auto input0 = model->addOperand(&type4); in CreateModel_quant8() local
70 model->addOperation(ANEURALNETWORKS_GATHER, {input0, param, indices}, {output0}); in CreateModel_quant8()
73 {input0, indices}, in CreateModel_quant8()
89 auto input0 = model->addOperand(&type6); in CreateModel_int32() local
[all …]
|
| /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 | 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 | 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 | 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(
104 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,
|