| /frameworks/ml/nn/runtime/test/specs/V1_2/ |
| D | depthwise_conv2d_dilation.mod.py | 43 }).AddNchw(i1, o1, layout).AddInput(f1, b1).AddVariations("relaxed", "float16", quant8)
71 }).AddNchw(i2, o2, layout).AddInput(f2, b2)
98 }, name="valid_padding").AddNchw(i1, o1, layout).AddInput(f1, b1).AddVariations("relaxed", "float16…
126 }, name="valid_padding").AddNchw(i2, o2, layout).AddInput(f2, b2)
154 }, name="same_padding_stride_2").AddNchw(i2, o2, layout).AddInput(f2, b2).AddVariations("relaxed", …
|
| D | conv2d_dilation.mod.py | 38 }).AddNchw(i1, o1, layout).AddInput(f1, b1).AddVariations("relaxed", quant8, "float16")
68 }).AddNchw(i2, o2, layout).AddInput(f2, b2).AddVariations("relaxed", quant8, "float16")
89 }, name="valid_padding").AddNchw(i1, o1, layout).AddInput(f1, b1).AddVariations("relaxed", quant8, …
119 }, name="valid_padding").AddNchw(i2, o2, layout).AddInput(f2, b2).AddVariations("relaxed", quant8, …
146 }).AddNchw(i3, o3, layout).AddInput(f3, b3).AddVariations("relaxed", quant8, "float16")
|
| D | depthwise_conv2d_v1_2.mod.py | 55 }).AddNchw(i1, o1, layout).AddInput(f1, b1).AddVariations("relaxed", "float16", channelQuant8, chan…
83 }).AddNchw(i2, o2, layout).AddInput(f2, b2).AddVariations("relaxed", "float16", quant8, channelQuan…
111 }).AddNchw(i3, o3, layout).AddInput(f3, b3).AddVariations("relaxed", "float16", quant8, channelQuan…
142 }).AddNchw(i4, o4, layout).AddInput(f4, b4).AddVariations("relaxed", "float16", quant8, channelQuan…
167 }, model=model9).AddInput(f9, b9).AddVariations("relaxed")
|
| D | depthwise_conv2d_per_channel.mod.py | 29 }).AddInput(f1, b1)
46 }).AddInput(f2, b2)
65 }).AddNchw(i3, o3, layout).AddInput(f3, b3)
|
| D | transpose_conv2d.mod.py | 65 …xed", quant8, quant8_mult_gt_1, channelQuant8, channelQuant8_mult_gt_1, "float16").AddInput(w1, b1)
97 }).AddNchw(i2, o2, s2, layout).AddVariations("relaxed", quant8, channelQuant8, "float16").AddInput(…
123 }).AddNchw(i3, o3, s3, layout).AddVariations("relaxed", quant8, "float16").AddInput(w3, b3)
151 }).AddNchw(i4, o4, s4, layout).AddVariations("relaxed", quant8, "float16").AddInput(w4, b4)
175 }).AddNchw(i5, o5, layout).AddVariations("relaxed", quant8, "float16").AddInput(w5, b5)
284 }).AddNchw(i8, o8, s8, layout).AddVariations("relaxed", quant8, "float16").AddInput(w8, b8)
|
| D | grouped_conv2d.mod.py | 65 …xed", quant8, quant8_mult_gt_1, channelQuant8, channelQuant8_mult_gt_1, "float16").AddInput(w1, b1)
101 }).AddNchw(i2, o2, layout).AddVariations("relaxed", quant8, channelQuant8, "float16").AddInput(w2, …
135 }).AddNchw(i3, o3, layout).AddVariations("relaxed", quant8, channelQuant8, "float16").AddInput(w3, …
|
| D | conv2d_v1_2.mod.py | 44 }).AddNchw(i1, o1, layout).AddInput(f1, b1).AddVariations("relaxed", quant8, channelQuant8, "float1…
72 }).AddNchw(i2, o2, layout).AddInput(f2, b2).AddVariations("relaxed", quant8, channelQuant8, "float1…
100 }).AddNchw(i3, o3, layout).AddInput(f3, b3).AddVariations("relaxed", quant8, channelQuant8, "float1…
140 }).AddNchw(i4, o4, layout).AddInput(f4, b4).AddVariations("relaxed", quant8, channelQuant8, channel…
211 }, model=model9).AddInput(f9, b9).AddVariations("relaxed")
|
| D | conv2d_per_channel.mod.py | 28 }).AddInput(f1, b1)
43 }).AddNchw(i2, o2, layout).AddInput(f2, b2)
|
| D | prelu.mod.py | 61 }).AddInput(a1).AddVariations("relaxed", quant8_gt, quant8_eq1, quant8_eq2, quant8_lt, "float16")
|
| /frameworks/ml/nn/common/operations/ |
| D | SVDFTest.cpp | 204 #define AddInput(X) \ in SVDFOpModel() macro
208 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(AddInput); in SVDFOpModel()
210 #undef AddInput in SVDFOpModel()
|
| D | LayerNormLSTMTest.cpp | 106 #define AddInput(X) \ in LayerNormLSTMOpModel() macro
111 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(AddInput); in LayerNormLSTMOpModel()
121 FOR_ALL_LAYER_NORM_WEIGHTS(AddInput); in LayerNormLSTMOpModel()
|
| D | LSTMTest.cpp | 96 #define AddInput(X) \ in LSTMOpModel() macro
100 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(AddInput); in LSTMOpModel()
|
| /frameworks/ml/nn/tools/test_generator/tests/P_variation/ |
| D | conv_float.mod.py | 44 output, act).AddInput(f1).AddVariations(RelaxedModeConverter(True), quant8)
|
| /frameworks/ml/nn/tools/test_generator/tests/P_vts_variation/ |
| D | conv_float.mod.py | 46 output, act).AddInput(f1).AddVariations(RelaxedModeConverter(True), quant8)
|
| /frameworks/ml/nn/tools/test_generator/ |
| D | README.md | 291 example.AddInput(*op_list)
371 example.AddInput(f1, b1).AddVariations("relaxed", quant8).AddAllActivations(o1, act)
|
| D | test_generator.py | 1041 def AddInput(self, *args, includeDefault=True, defaultName=None): member in Example
|