Home
last modified time | relevance | path

Searched refs:AddNchw (Results 1 – 25 of 27) sorted by relevance

12

/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 …]
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")
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")
254 }).AddNchw(i1, zero_sized, o3, layout).AddVariations("relaxed", quant8, "float16")
297 }).AddNchw(i1, zero_sized, o3, layout).AddVariations("relaxed", quant8, "float16")
Ddepthwise_conv2d_dilation.mod.py43 }).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", …
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)
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…
127 }).AddNchw(i1, zero_sized, o3, layout).AddVariations("relaxed", quant8, "float16")
166 }).AddNchw(i1, zero_sized, o3, layout).AddVariations("relaxed", quant8, "float16")
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")
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)
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…
Dinstance_normalization.mod.py36 }).AddNchw(i1, o1, layout).AddVariations("relaxed", "float16")
56 }).AddNchw(i2, o2, layout).AddVariations("relaxed", "float16")
Dbatch_to_space_v1_2.mod.py34 }).AddNchw(i1, o1, layout).AddVariations("relaxed", "float16", quant8)
52 }).AddNchw(i2, o2, layout).AddVariations("relaxed", "float16", quant8)
Dgrouped_conv2d.mod.py65 }).AddNchw(i1, o1, layout).AddAllActivations(o1, act).AddVariations("relaxed", quant8, quant8_mult_…
101 }).AddNchw(i2, o2, layout).AddVariations("relaxed", quant8, channelQuant8, "float16").AddInput(w2, …
135 }).AddNchw(i3, o3, layout).AddVariations("relaxed", quant8, channelQuant8, "float16").AddInput(w3, …
Dgenerate_proposals.mod.py66 Example((input0, output0)).AddNchw(i1, i2, layout).AddVariations("relaxed", quant8, "float16")
211 Example((input0, output0)).AddNchw(i1, i2, layout).AddVariations("relaxed", quant8, "float16")
Dconv2d_per_channel.mod.py43 }).AddNchw(i2, o2, layout).AddInput(f2, b2)
74 }).AddNchw(i1, zero_sized, o3, layout)
Ddepthwise_conv2d_per_channel.mod.py65 }).AddNchw(i3, o3, layout).AddInput(f3, b3)
Dfully_connected_v1_2.mod.py86 }).AddNchw(i1, zero_sized, layout).AddVariations("relaxed", quant8, "float16")
/frameworks/ml/nn/tools/test_generator/tests/P_variation/
Dconv_float.mod.py43 Example((input0, output0)).AddNchw(i1, f1, output, layout).AddAllActivations(
/frameworks/ml/nn/tools/test_generator/tests/P_vts_variation/
Dconv_float.mod.py45 Example((input0, output0)).AddNchw(i1, f1, output, layout).AddAllActivations(

12