Home
last modified time | relevance | path

Searched refs:o2 (Results 1 – 25 of 86) sorted by relevance

1234

/frameworks/ml/nn/runtime/test/specs/V1_2/
Ddepthwise_conv2d_dilation.mod.py50 o2 = Output("op4", "TENSOR_FLOAT32", "{1, 2, 2, 4}") variable
51 Model().Operation("DEPTHWISE_CONV_2D", i2, f2, b2, 0, 0, 0, 0, 1, 1, 2, 0, layout, 2, 2).To(o2)
58 o2: ("TENSOR_QUANT8_ASYMM", 0.1, 0)
67 o2: [13, 14, 0, 0,
71 }).AddNchw(i2, o2, layout).AddInput(f2, b2)
105 o2 = Output("op4", "TENSOR_FLOAT32", "{1, 2, 2, 4}") variable
106 Model().Operation("DEPTHWISE_CONV_2D", i2, f2, b2, 2, 1, 1, 2, 0, layout, 2, 2).To(o2)
113 o2: ("TENSOR_QUANT8_ASYMM", 0.1, 0)
122 o2: [13, 14, 0, 0,
126 }, name="valid_padding").AddNchw(i2, o2, layout).AddInput(f2, b2)
[all …]
Ddetection_postprocess.mod.py23 o2 = Output("roiOut", "TENSOR_FLOAT32", "{1, 3, 4}") # roi out variable
26 …STPROCESSING", i1, i2, i3, 10.0, 10.0, 5.0, 5.0, True, 3, 1, 1, 0.0, 0.5, False).To(o1, o2, o3, o4)
57 o2: [
74 o2 = Output("roiOut", "TENSOR_FLOAT32", "{1, 3, 4}") # roi out variable
77 …TPROCESSING", i1, i2, i3, 10.0, 10.0, 5.0, 5.0, False, 3, 1, 1, 0.0, 0.5, False).To(o1, o2, o3, o4)
108 o2: [
125 o2 = Output("roiOut", "TENSOR_FLOAT32", "{1, 3, 4}") # roi out variable
128 …TPROCESSING", i1, i2, i3, 10.0, 10.0, 5.0, 5.0, False, 3, 1, 1, 0.0, 0.5, False).To(o1, o2, o3, o4)
159 o2: [
176 o2 = Output("roiOut", "TENSOR_FLOAT32", "{1, 3, 4}") # roi out variable
[all …]
Dbox_with_nms_limit_linear.mod.py23 o2 = Output("roiOut", "TENSOR_FLOAT32", "{16, 4}") # roi out variable
26 model = Model().Operation("BOX_WITH_NMS_LIMIT", i1, i2, i3, 0.3, -1, 1, 0.4, 1.0, 0.3).To(o1, o2, o…
32 o2: ("TENSOR_QUANT16_ASYMM", 0.125, 0)
86 o2: [
117 o2 = Output("roiOut", "TENSOR_FLOAT32", "{15, 4}") # roi out variable
120 model = Model().Operation("BOX_WITH_NMS_LIMIT", i1, i2, i3, 0.3, 8, 1, 0.4, 0.5, 0.3).To(o1, o2, o3…
126 o2: ("TENSOR_QUANT16_ASYMM", 0.125, 0)
180 o2: [
Dbox_with_nms_limit_gaussian.mod.py23 o2 = Output("roiOut", "TENSOR_FLOAT32", "{18, 4}") # roi out variable
26 model = Model().Operation("BOX_WITH_NMS_LIMIT", i1, i2, i3, 0.3, -1, 2, 0.4, 0.5, 0.3).To(o1, o2, o…
32 o2: ("TENSOR_QUANT16_ASYMM", 0.125, 0)
86 o2: [
119 o2 = Output("roiOut", "TENSOR_FLOAT32", "{10, 4}") # roi out variable
122 model = Model().Operation("BOX_WITH_NMS_LIMIT", i1, i2, i3, 0.3, 5, 2, 0.4, 0.5, 0.3).To(o1, o2, o3…
128 o2: ("TENSOR_QUANT16_ASYMM", 0.125, 0)
182 o2: [
Dbox_with_nms_limit_hard.mod.py23 o2 = Output("roiOut", "TENSOR_FLOAT32", "{12, 4}") # roi out variable
26 model = Model().Operation("BOX_WITH_NMS_LIMIT", i1, i2, i3, 0.3, -1, 0, 0.4, 1.0, 0.3).To(o1, o2, o…
32 o2: ("TENSOR_QUANT16_ASYMM", 0.125, 0)
83 o2: [
110 o2 = Output("roiOut", "TENSOR_FLOAT32", "{10, 4}") # roi out variable
113 model = Model().Operation("BOX_WITH_NMS_LIMIT", i1, i2, i3, 0.3, 5, 0, 0.4, 0.5, 0.3).To(o1, o2, o3…
119 o2: ("TENSOR_QUANT16_ASYMM", 0.125, 0)
170 o2: [
Dresize_bilinear_v1_2.mod.py45 o2 = Output("op4", "TENSOR_FLOAT32", "{1, 3, 3, 2}") variable
46 model_shape = Model("shape").Operation("RESIZE_BILINEAR", i2, 3, 3, layout).To(o2)
47 model_scale = Model("scale").Operation("RESIZE_BILINEAR", i2, 1.6, 1.6, layout).To(o2)
52 o2: ("TENSOR_QUANT8_ASYMM", 0.25, 0)
57 o2: [3, 4, 5, 8, 6, 10,
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…
97 o2 = Output("classesOut", "TENSOR_INT32", "{0}") # classes out variable
100 …d").Operation("BOX_WITH_NMS_LIMIT", p1, p2, [0], 0.3, -1, 0, 0.4, 1.0, 0.3).To(o1, tmp1, o2, tmp2)
125 o2: [0],
[all …]
Dl2_pool_v1_2.mod.py33 o2 = Output("op4", "TENSOR_FLOAT32", "{1, 1, 2, 1}") variable
34 Model().Operation("L2_POOL_2D", i2, 1, 2, 2, 2, 2, 0, layout).To(o2)
39 o2: [3.5, 6.5]
40 }).AddNchw(i2, o2, layout).AddRelaxed().AddVariations("float16")
61 o2 = Output("classesOut", "TENSOR_INT32", "{0}") # classes out variable
64 …d").Operation("BOX_WITH_NMS_LIMIT", p1, p2, [0], 0.3, -1, 0, 0.4, 1.0, 0.3).To(o1, tmp1, o2, tmp2)
79 o2: [0],
90 o2 = Output("classesOut", "TENSOR_INT32", "{0}") # classes out variable
93 …d").Operation("BOX_WITH_NMS_LIMIT", p1, p2, [0], 0.3, -1, 0, 0.4, 1.0, 0.3).To(o1, tmp1, o2, tmp2)
108 o2: [0],
Dconv2d_dilation.mod.py45 o2 = Output("op4", "TENSOR_FLOAT32", "{1, 3, 3, 1}") variable
46 Model().Operation("CONV_2D", i2, f2, b2, 0, 0, 0, 0, 1, 1, 0, layout, 3, 3).To(o2)
53 o2: ("TENSOR_QUANT8_ASYMM", 0.125, 0)
67 o2: [5, 5, 5, 5, 5, 5, 5, 5, 5]
68 }).AddNchw(i2, o2, layout).AddInput(f2, b2).AddVariations("relaxed", quant8, "float16")
96 o2 = Output("op4", "TENSOR_FLOAT32", "{1, 3, 3, 1}") variable
97 Model().Operation("CONV_2D", i2, f2, b2, 2, 1, 1, 0, layout, 3, 3).To(o2)
104 o2: ("TENSOR_QUANT8_ASYMM", 0.125, 0)
118 o2: [5, 5, 5, 5, 5, 5, 5, 5, 5]
119 }, name="valid_padding").AddNchw(i2, o2, layout).AddInput(f2, b2).AddVariations("relaxed", quant8, …
Dgenerate_proposals.mod.py26 o2 = Output("roiOut", "TENSOR_FLOAT32", "{4, 4}") # roi out variable
29 i1, i2, i3, i4, 4.0, 4.0, -1, -1, 0.30, 1.0, layout).To(o1, o2, o3)
37 o2: ("TENSOR_QUANT16_ASYMM", 0.125, 0)
57 o2: [ # roi out
75 o2 = Output("roiOut", "TENSOR_FLOAT32", "{30, 4}") # roi out variable
78 i1, i2, i3, i4, 10.0, 10.0, 32, 16, 0.20, 1.0, layout).To(o1, o2, o3)
86 o2: ("TENSOR_QUANT16_ASYMM", 0.125, 0)
173 o2: [ # roi out
Dconv2d_per_channel.mod.py36 o2 = Output("op4", "TENSOR_QUANT8_ASYMM", "{1, 3, 1, 3}, 1.f, 128") variable
37 Model("layouts").Operation("CONV_2D", i2, f2, b2, 0, 0, 0, 0, 1, 1, 0, layout).To(o2)
42 o2: [121, 118, 115, 121, 118, 115, 121, 118, 115]
43 }).AddNchw(i2, o2, layout).AddInput(f2, b2)
51 o2 = Output("classesOut", "TENSOR_INT32", "{0}") # classes out variable
54 …ed").Operation("BOX_WITH_NMS_LIMIT", p1, p2, [0], 0.3, -1, 0, 0.4, 1.0, 0.3).To(o1, tmp1, o2, tmp2)
72 o2: [0],
Dmax_pool_v1_2.mod.py49 o2 = Output("op4", ("TENSOR_FLOAT32", [bat, output_row, output_col, chn])) variable
50 Model().Operation("MAX_POOL_2D", i2, pad, pad, pad, pad, std, std, flt, flt, 0, layout).To(o2)
55 o2: ("TENSOR_QUANT8_ASYMM", 0.5, 0)
61 o2: [std for _ in range(bat * output_row * output_col * chn)]
62 }).AddNchw(i2, o2, layout).AddVariations("relaxed", quant8, "float16")
117 o2 = Output("classesOut", "TENSOR_INT32", "{0}") # classes out variable
120 …d").Operation("BOX_WITH_NMS_LIMIT", p1, p2, [0], 0.3, -1, 0, 0.4, 1.0, 0.3).To(o1, tmp1, o2, tmp2)
145 o2: [0],
156 o2 = Output("classesOut", "TENSOR_INT32", "{0}") # classes out variable
159 …d").Operation("BOX_WITH_NMS_LIMIT", p1, p2, [0], 0.3, -1, 0, 0.4, 1.0, 0.3).To(o1, tmp1, o2, tmp2)
[all …]
Dbatch_to_space_v1_2.mod.py39 o2 = Output("op4", "TENSOR_FLOAT32", "{1, 4, 4, 1}") variable
40 Model().Operation("BATCH_TO_SPACE_ND", i2, [2, 2], layout).To(o2)
45 o2: ("TENSOR_QUANT8_ASYMM", 0.5, 128)
51 o2: [1, 5, 2, 6, 9, 13, 10, 14, 3, 7, 4, 8, 11, 15, 12, 16]
52 }).AddNchw(i2, o2, layout).AddVariations("relaxed", "float16", quant8)
Droi_align.mod.py57 o2 = Output("out", "TENSOR_FLOAT32", "{4, 2, 3, 2}") variable
58 Model().Operation("ROI_ALIGN", i2, roi2, [0, 0, 3, 3], 2, 3, 4.0, 4.0, 4, 4, layout).To(o2)
63 o2: ("TENSOR_QUANT8_ASYMM", 0.03125, 10)
104 o2: [
118 }).AddNchw(i2, o2, layout).AddVariations("relaxed", quant8, "float16")
220 o2 = Output("classesOut", "TENSOR_INT32", "{0}") # classes out variable
223 …d").Operation("BOX_WITH_NMS_LIMIT", p1, p2, [0], 0.3, -1, 0, 0.4, 1.0, 0.3).To(o1, tmp1, o2, tmp2)
243 o2: [0],
Davg_pool_v1_2.mod.py49 o2 = Output("op4", ("TENSOR_FLOAT32", [bat, output_row, output_col, chn])) variable
50 Model().Operation("AVERAGE_POOL_2D", i2, pad, pad, pad, pad, std, std, flt, flt, 0, layout).To(o2)
55 o2: ("TENSOR_QUANT8_ASYMM", 0.5, 0)
61 o2: [1. for _ in range(bat * output_row * output_col * chn)]
62 }).AddNchw(i2, o2, layout).AddVariations("relaxed", "float16", quant8)
145 o2 = Output("classesOut", "TENSOR_INT32", "{0}") # classes out variable
148 …ed").Operation("BOX_WITH_NMS_LIMIT", p1, p2, [0], 0.3, -1, 0, 0.4, 1.0, 0.3).To(o1, tmp1, o2, tmp2)
173 o2: [0],
184 o2 = Output("classesOut", "TENSOR_INT32", "{0}") # classes out variable
187 …ed").Operation("BOX_WITH_NMS_LIMIT", p1, p2, [0], 0.3, -1, 0, 0.4, 1.0, 0.3).To(o1, tmp1, o2, tmp2)
[all …]
Dtranspose_conv2d.mod.py73 o2 = Output("op4", "TENSOR_FLOAT32", "{1, 3, 4, 1}") # output variable
74 Model().Operation("TRANSPOSE_CONV_2D", i2, w2, b2, s2, 1, 3, 3, 1, layout).To(o2)
81 o2: ("TENSOR_QUANT8_ASYMM", 20.0, 50)
89 o2: ("TENSOR_QUANT8_ASYMM", 20.0, 50)
94 o2: [500., 800., 3500., 1500.,
97 }).AddNchw(i2, o2, s2, layout).AddVariations("relaxed", quant8, channelQuant8, "float16").AddInput(…
184 o2 = Output("classesOut", "TENSOR_INT32", "{0}") # classes out variable
187 …ed").Operation("BOX_WITH_NMS_LIMIT", p1, p2, [0], 0.3, -1, 0, 0.4, 1.0, 0.3).To(o1, tmp1, o2, tmp2)
217 o2: [0],
228 o2 = Output("classesOut", "TENSOR_INT32", "{0}") # classes out variable
[all …]
Dconcat_zero_sized.mod.py23 o2 = Output("classesOut", "TENSOR_INT32", "{0}") # classes out variable
26 …el().Operation("BOX_WITH_NMS_LIMIT", p1, p2, [0], 0.3, -1, 0, 0.4, 1.0, 0.3).To(o1, tmp1, o2, tmp2)
52 o2: [0],
63 o2 = Output("classesOut", "TENSOR_INT32", "{0}") # classes out variable
66 …el().Operation("BOX_WITH_NMS_LIMIT", p1, p2, [0], 0.3, -1, 0, 0.4, 1.0, 0.3).To(o1, tmp1, o2, tmp2)
95 o2: [0],
Dspace_to_depth_v1_2.mod.py39 o2 = Output("op4", "TENSOR_FLOAT32", "{1, 2, 2, 4}") variable
40 Model().Operation("SPACE_TO_DEPTH", i2, 2, layout).To(o2)
45 o2: ("TENSOR_QUANT8_ASYMM", 0.5, 128)
51 o2: [1., 2., 3., 4., 5., 6., 7., 8., 9., 10., 11., 12., 13., 14., 15., 16.]
52 }).AddNchw(i2, o2, layout).AddVariations("relaxed", "float16", quant8)
Ddepth_to_space_v1_2.mod.py39 o2 = Output("op4", "TENSOR_FLOAT32", "{1, 4, 4, 1}") variable
40 Model().Operation("DEPTH_TO_SPACE", i2, 2, layout).To(o2)
45 o2: ("TENSOR_QUANT8_ASYMM", 0.5, 128)
51 o2: [1., 2., 3., 4., 5., 6., 7., 8., 9., 10., 11., 12., 13., 14., 15., 16.]
52 }).AddNchw(i2, o2, layout).AddVariations("relaxed", "float16", quant8)
Droi_pooling.mod.py59 o2 = Output("out", "TENSOR_FLOAT32", "{4, 2, 3, 2}") variable
60 Model().Operation("ROI_POOLING", i2, roi2, [0, 0, 3, 3], 2, 3, 4.0, 4.0, layout).To(o2)
65 o2: ("TENSOR_QUANT8_ASYMM", 0.04, 0)
106 o2: [
112 }).AddNchw(i2, o2, layout).AddVariations("relaxed", quant8, "float16")
Dinstance_normalization.mod.py41 o2 = Output("out", "TENSOR_FLOAT32", "{2, 2, 2, 2}") variable
42 Model().Operation("INSTANCE_NORMALIZATION", i2, 2.0, 10.0, 0.0001, layout).To(o2)
50 o2: [
56 }).AddNchw(i2, o2, layout).AddVariations("relaxed", "float16")
Dconv2d_v1_2.mod.py51 o2 = Output("op4", "TENSOR_FLOAT32", "{1, 3, 4, 1}") variable
52 Model().Operation("CONV_2D", i2, f2, b2, 1, 1, 1, 1, layout).To(o2)
59 o2: ("TENSOR_QUANT8_ASYMM", 1.0, 50)
65 o2: ("TENSOR_QUANT8_ASYMM", 1.0, 50)
71 o2: [0, 0, 0, 0, 35, 112, 157, 0, 0, 34, 61, 0]
72 }).AddNchw(i2, o2, layout).AddInput(f2, b2).AddVariations("relaxed", quant8, channelQuant8, "float1…
220 o2 = Output("classesOut", "TENSOR_INT32", "{0}") # classes out variable
223 …ed").Operation("BOX_WITH_NMS_LIMIT", p1, p2, [0], 0.3, -1, 0, 0.4, 1.0, 0.3).To(o1, tmp1, o2, tmp2)
252 o2: [0],
263 o2 = Output("classesOut", "TENSOR_INT32", "{0}") # classes out variable
[all …]
Dspace_to_batch_v1_2.mod.py40 o2 = Output("op4", "TENSOR_FLOAT32", "{4, 2, 2, 1}") variable
41 Model().Operation("SPACE_TO_BATCH_ND", i2, [2, 2], pad1, layout).To(o2)
46 o2: ("TENSOR_QUANT8_ASYMM", 0.5, 0)
52 o2: [1, 3, 9, 11, 2, 4, 10, 12, 5, 7, 13, 15, 6, 8, 14, 16]
53 }).AddNchw(i2, o2, layout).AddVariations("relaxed", "float16", quant8)
Dgrouped_conv2d.mod.py72 o2 = Output("op4", "TENSOR_FLOAT32", "{1, 3, 2, 2}") # output 0 variable
73 Model("large").Operation("GROUPED_CONV_2D", i2, w2, b2, 1, 1, 1, 2, 0, layout).To(o2)
80 o2: ("TENSOR_QUANT8_ASYMM", 10.0, 100)
88 o2: ("TENSOR_QUANT8_ASYMM", 10.0, 100)
95 o2: [567, -873,
101 }).AddNchw(i2, o2, layout).AddVariations("relaxed", quant8, channelQuant8, "float16").AddInput(w2, …
/frameworks/ml/nn/tools/test_generator/tests/P_implicit_parameter/
Dmean_implicit.mod.py18 o2 = Output("o2", ("TENSOR_FLOAT32", [2, 1])) # output for model2 variable
22 model2 = Model().Operation("MEAN", i0, [1], 1).To(o2) # along axis 1, keep_dim=True
27 outputs12 = {o2: [3, 7]}
32 outputs22 = {o2: [-3, -7]}
/frameworks/ml/nn/tools/test_generator/tests/P_vts_implicit_parameter/
Dmean_implicit.mod.py18 o2 = Output("o2", ("TENSOR_FLOAT32", [2, 1])) # output for model2 variable
22 model2 = Model().Operation("MEAN", i0, [1], 1).To(o2) # along axis 1, keep_dim=True
27 outputs12 = {o2: [3, 7]}
32 outputs22 = {o2: [-3, -7]}

1234