Home
last modified time | relevance | path

Searched refs:AddVariations (Results 1 – 25 of 108) sorted by relevance

12345

/frameworks/ml/nn/runtime/test/specs/V1_2/
Dresize_nearest_neighbor.mod.py36 Example(test1, model=model_shape).AddNchw(i1, o1, layout).AddVariations("relaxed", quant8, "float16…
37 Example(test1, model=model_scale).AddNchw(i1, o1, layout).AddVariations("relaxed", quant8, "float16…
57 Example(test2, model=model_shape).AddNchw(i1, o1, layout).AddVariations("relaxed", quant8, "float16…
58 Example(test2, model=model_scale).AddNchw(i1, o1, layout).AddVariations("relaxed", quant8, "float16…
78 Example(test3, model=model_shape).AddNchw(i1, o1, layout).AddVariations("relaxed", quant8, "float16…
79 Example(test3, model=model_scale).AddNchw(i1, o1, layout).AddVariations("relaxed", quant8, "float16…
99 Example(test4, model=model_shape).AddNchw(i1, o1, layout).AddVariations("relaxed", quant8, "float16…
100 Example(test4, model=model_scale).AddNchw(i1, o1, layout).AddVariations("relaxed", quant8, "float16…
120 Example(test5, model=model_shape).AddNchw(i1, o1, layout).AddVariations("relaxed", quant8, "float16…
121 Example(test5, model=model_scale).AddNchw(i1, o1, layout).AddVariations("relaxed", quant8, "float16…
[all …]
Dresize_bilinear_v1_2.mod.py39 Example(test1, model=model_shape).AddNchw(i1, o1, layout).AddVariations("relaxed", "float16", quant…
40 Example(test1, model=model_scale).AddNchw(i1, o1, layout).AddVariations("relaxed", "float16", quant…
63 Example(test2, model=model_shape).AddNchw(i2, o2, layout).AddVariations("relaxed", "float16", quant…
64 Example(test2, model=model_scale).AddNchw(i2, o2, layout).AddVariations("relaxed", "float16", quant…
87 Example(test3, model=model_shape).AddVariations("float16", quant8, includeDefault=False)
88 Example(test3, model=model_scale).AddVariations("float16", quant8, includeDefault=False)
127 }).AddNchw(i1, zero_sized, o3, layout).AddVariations("relaxed", quant8, "float16")
166 }).AddNchw(i1, zero_sized, o3, layout).AddVariations("relaxed", quant8, "float16")
Dconv2d_v1_2.mod.py44 }).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…
158 }, model=model_1_same).AddNchw(i5, o5, layout).AddVariations("relaxed", "float16")
166 }, model=model_1_valid).AddNchw(i5, o6, layout).AddVariations("relaxed", "float16")
184 }, model=model_3_same).AddNchw(i7, o7, layout).AddVariations("relaxed", "float16")
192 }, model=model_3_valid).AddNchw(i7, o8, layout).AddVariations("relaxed", "float16")
211 }, model=model9).AddInput(f9, b9).AddVariations("relaxed")
254 }).AddNchw(i1, zero_sized, o3, layout).AddVariations("relaxed", quant8, "float16")
[all …]
Dsoftmax_v1_2.mod.py44 Example(example1).AddVariations("relaxed", "float16", quant8).AddDims([1, 3], i, o)
47 Example(example2).AddVariations("relaxed", "float16", quant8).AddDims([1, 3], i, o)
52 Example(example1).AddVariations("relaxed", "float16", quant8).AddAllDimsAndAxis(i, o, axis)
55 Example(example2).AddVariations("relaxed", "float16", quant8).AddAllDimsAndAxis(i, o, axis)
99 }).AddVariations("relaxed", quant8, "float16")
Dlocal_response_normalization_v1_2.mod.py26 })).AddRelaxed().AddAllDimsAndAxis(i, o, axis).AddVariations("float16")
32 })).AddRelaxed().AddAllDimsAndAxis(i, o, axis).AddVariations("float16")
38 })).AddRelaxed().AddAllDimsAndAxis(i, o, axis).AddVariations("float16")
44 })).AddRelaxed().AddDims([1, 2, 3], i, o, includeDefault=False).AddVariations("float16")
Droi_align.mod.py51 }).AddNchw(i1, o1, layout).AddVariations("relaxed", quant8, "float16")
118 }).AddNchw(i2, o2, layout).AddVariations("relaxed", quant8, "float16")
171 }).AddNchw(i3, o3, layout).AddVariations("relaxed", quant8, "float16")
211 }).AddNchw(i4, o4, layout).AddVariations("relaxed", quant8, "float16")
245 }).AddNchw(i1, zero_sized, layout).AddVariations("relaxed", quant8, "float16")
265 }).AddNchw(i4, o4, layout).AddVariations("relaxed", quant8, "float16")
Dl2_pool_v1_2.mod.py28 }).AddNchw(i1, o1, layout).AddRelaxed().AddVariations("float16")
40 }).AddNchw(i2, o2, layout).AddRelaxed().AddVariations("float16")
52 }).AddNchw(i3, o3, layout).AddRelaxed().AddVariations("float16")
81 }).AddNchw(i1, zero_sized, o3, layout).AddVariations("relaxed", "float16")
110 }).AddNchw(i1, zero_sized, o3, layout).AddVariations("relaxed", "float16")
Dconv2d_dilation.mod.py38 }).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")
Dspace_to_batch_v1_2.mod.py35 }).AddNchw(i1, o1, layout).AddVariations("relaxed", "float16", quant8)
53 }).AddNchw(i2, o2, layout).AddVariations("relaxed", "float16", quant8)
73 }).AddNchw(i3, o3, layout).AddVariations("relaxed", "float16", quant8)
94 }).AddNchw(i4, o4, layout).AddVariations("relaxed", "float16", quant8)
Dtranspose_conv2d.mod.py65 }).AddNchw(i1, o1, s1, layout).AddAllActivations(o1, act).AddVariations("relaxed", quant8, quant8_m…
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)
219 }).AddNchw(i1, zero_sized, o3, s, layout).AddVariations("relaxed", quant8, "float16")
262 }).AddNchw(i1, zero_sized, o3, layout).AddVariations("relaxed", quant8, "float16")
284 }).AddNchw(i8, o8, s8, layout).AddVariations("relaxed", quant8, "float16").AddInput(w8, b8)
Davg_pool_v1_2.mod.py34 }).AddNchw(i1, o1, layout).AddVariations("relaxed", "float16", quant8)
62 }).AddNchw(i2, o2, layout).AddVariations("relaxed", "float16", quant8)
90 }).AddNchw(i3, o3, layout).AddVariations("relaxed", "float16", quant8)
118 }).AddNchw(i4, o4, layout).AddVariations("relaxed", "float16", quant8)
136 }).AddNchw(i5, o5, layout).AddVariations("relaxed", "float16", quant8)
175 }).AddNchw(i1, zero_sized, o3, layout).AddVariations("relaxed", quant8, "float16")
214 }).AddNchw(i1, zero_sized, o3, layout).AddVariations("relaxed", quant8, "float16")
Dmax_pool_v1_2.mod.py34 }).AddNchw(i1, o1, layout).AddVariations("relaxed", quant8, "float16")
62 }).AddNchw(i2, o2, layout).AddVariations("relaxed", quant8, "float16")
90 }).AddNchw(i3, o3, layout).AddVariations("relaxed", quant8, "float16")
108 }).AddNchw(i4, o4, layout).AddVariations("relaxed", quant8, "float16")
147 }).AddNchw(i1, zero_sized, o3, layout).AddVariations("relaxed", quant8, "float16")
186 }).AddNchw(i1, zero_sized, o3, layout).AddVariations("relaxed", quant8, "float16")
Ddetection_postprocess.mod.py66 Example((input0, output0)).AddVariations("relaxed", "float16")
117 Example((input0, output0)).AddVariations("relaxed", "float16")
168 Example((input0, output0)).AddVariations("relaxed", "float16")
219 Example((input0, output0)).AddVariations("relaxed", "float16")
Ddepthwise_conv2d_v1_2.mod.py55 }).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")
Dspace_to_depth_v1_2.mod.py34 }).AddNchw(i1, o1, layout).AddVariations("relaxed", "float16", quant8)
52 }).AddNchw(i2, o2, layout).AddVariations("relaxed", "float16", quant8)
76 }).AddNchw(i3, o3, layout).AddVariations("relaxed", "float16", quant8)
Ddepth_to_space_v1_2.mod.py34 }).AddNchw(i1, o1, layout).AddVariations("relaxed", "float16", quant8)
52 }).AddNchw(i2, o2, layout).AddVariations("relaxed", "float16", quant8)
76 }).AddNchw(i3, o3, layout).AddVariations("relaxed", "float16", quant8)
Droi_pooling.mod.py53 }).AddNchw(i1, o1, layout).AddVariations("relaxed", quant8, "float16")
112 }).AddNchw(i2, o2, layout).AddVariations("relaxed", quant8, "float16")
152 }).AddNchw(i3, o3, layout).AddVariations("relaxed", quant8, "float16")
Dheatmap_max_keypoint.mod.py78 }).AddNchw(heatmap1, layout).AddVariations("relaxed", "float16")
131 }).AddNchw(heatmap2, layout).AddVariations("relaxed", "float16", quant8)
193 }).AddNchw(heatmap3, layout).AddVariations(quant8, includeDefault=False)
Dinstance_normalization.mod.py36 }).AddNchw(i1, o1, layout).AddVariations("relaxed", "float16")
56 }).AddNchw(i2, o2, layout).AddVariations("relaxed", "float16")
Ddepthwise_conv2d_dilation.mod.py43 }).AddNchw(i1, o1, layout).AddInput(f1, b1).AddVariations("relaxed", "float16", quant8)
98 }, name="valid_padding").AddNchw(i1, o1, layout).AddInput(f1, b1).AddVariations("relaxed", "float16…
154 }, name="same_padding_stride_2").AddNchw(i2, o2, layout).AddInput(f2, b2).AddVariations("relaxed", …
/frameworks/ml/nn/tools/test_generator/
DREADME.md210 Use `AddVariations` to add a group of variations to the example
214 example.AddVariations(nchw).AddVariations(relaxed, quant8)
220 example.AddVariations(nchw, defaultName="nhwc").AddVariations(relaxed, quant8)
227 example.AddVariations(nchw, includeDefault=False).AddVariations(relaxed, quant8)
240 example.AddVariations(DataTypeConverter().Identify({op1: "TENSOR_FLOAT16", ...}))
241 example.AddVariations("float16") # will apply to every TENSOR_FLOAT32 operands
243 example.AddVariations(DataTypeConverter().Identify({op1: "TENSOR_INT32", ...}))
244 example.AddVariations("int32") # will apply to every TENSOR_FLOAT32 operands
247 example.AddVariations(DataLayoutConverter("nchw").Identify(op_list))
248 example.AddVariations(("nchw", op_list))
[all …]
/frameworks/ml/nn/tools/test_generator/tests/P_vts_implicit_variation/
Dconv_float.mod.py42 ).AddVariations(
44 ).AddVariations(
48 ).AddVariations(
50 ).AddVariations(
/frameworks/ml/nn/tools/test_generator/tests/P_implicit_variation/
Dconv_float.mod.py42 ).AddVariations(
44 ).AddVariations(
48 ).AddVariations(
50 ).AddVariations(
/frameworks/ml/nn/tools/test_generator/tests/P_vts_naming/
Dconv_float.mod.py53 ).AddVariations(
55 ).AddVariations(
57 ).AddVariations(
59 ).AddVariations(
/frameworks/ml/nn/tools/test_generator/tests/P_naming/
Dconv_float.mod.py53 ).AddVariations(
55 ).AddVariations(
57 ).AddVariations(
59 ).AddVariations(

12345