Skip to content
Snippets Groups Projects
Commit 0c43b60a authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Test] correct expected im2col values for inner dim ordering

parent 5592bc04
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
tests/custom_op/test_im2col.py
+ 37
− 37
View file @ 0c43b60a
......@@ -46,7 +46,7 @@ def execution_im2col(x, idt, k, stride, ifm_ch, ifm_dim, pad_amt=0, pad_val=0):
)
graph = helper.make_graph(
nodes=[Im2Col_node], name="im2col_graph", inputs=[inp], outputs=[outp],
nodes=[Im2Col_node], name="im2col_graph", inputs=[inp], outputs=[outp]
)
model = helper.make_model(graph, producer_name="im2col-model")
......@@ -175,19 +175,19 @@ def test_im2col():
[
[
[
[1.0, 2.0, 5.0, 6.0, -1.0, -2.0, -5.0, -6.0],
[2.0, 3.0, 6.0, 7.0, -2.0, -3.0, -6.0, -7.0],
[3.0, 4.0, 7.0, 8.0, -3.0, -4.0, -7.0, -8.0],
[1.0, -1.0, 2.0, -2.0, 5.0, -5.0, 6.0, -6.0],
[2.0, -2.0, 3.0, -3.0, 6.0, -6.0, 7.0, -7.0],
[3.0, -3.0, 4.0, -4.0, 7.0, -7.0, 8.0, -8.0],
],
[
[5.0, 6.0, 9.0, 10.0, -5.0, -6.0, -9.0, -10.0],
[6.0, 7.0, 10.0, 11.0, -6.0, -7.0, -10.0, -11.0],
[7.0, 8.0, 11.0, 12.0, -7.0, -8.0, -11.0, -12.0],
[5.0, -5.0, 6.0, -6.0, 9.0, -9.0, 10.0, -10.0],
[6.0, -6.0, 7.0, -7.0, 10.0, -10.0, 11.0, -11.0],
[7.0, -7.0, 8.0, -8.0, 11.0, -11.0, 12.0, -12.0],
],
[
[9.0, 10.0, 13.0, 14.0, -9.0, -10.0, -13.0, -14.0],
[10.0, 11.0, 14.0, 15.0, -10.0, -11.0, -14.0, -15.0],
[11.0, 12.0, 15.0, 16.0, -11.0, -12.0, -15.0, -16.0],
[9.0, -9.0, 10.0, -10.0, 13.0, -13.0, 14.0, -14.0],
[10.0, -10.0, 11.0, -11.0, 14.0, -14.0, 15.0, -15.0],
[11.0, -11.0, 12.0, -12.0, 15.0, -15.0, 16.0, -16.0],
],
]
],
......@@ -222,39 +222,39 @@ def test_im2col():
[
[
[
[0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, -1.0],
[0.0, 0.0, 1.0, 2.0, 0.0, 0.0, -1.0, -2.0],
[0.0, 0.0, 2.0, 3.0, 0.0, 0.0, -2.0, -3.0],
[0.0, 0.0, 3.0, 4.0, 0.0, 0.0, -3.0, -4.0],
[0.0, 0.0, 4.0, 0.0, 0.0, 0.0, -4.0, 0.0],
[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, -1.0],
[0.0, 0.0, 0.0, 0.0, 1.0, -1.0, 2.0, -2.0],
[0.0, 0.0, 0.0, 0.0, 2.0, -2.0, 3.0, -3.0],
[0.0, 0.0, 0.0, 0.0, 3.0, -3.0, 4.0, -4.0],
[0.0, 0.0, 0.0, 0.0, 4.0, -4.0, 0.0, 0.0],
],
[
[0.0, 1.0, 0.0, 5.0, 0.0, -1.0, 0.0, -5.0],
[1.0, 2.0, 5.0, 6.0, -1.0, -2.0, -5.0, -6.0],
[2.0, 3.0, 6.0, 7.0, -2.0, -3.0, -6.0, -7.0],
[3.0, 4.0, 7.0, 8.0, -3.0, -4.0, -7.0, -8.0],
[4.0, 0.0, 8.0, 0.0, -4.0, 0.0, -8.0, 0.0],
[0.0, 0.0, 1.0, -1.0, 0.0, 0.0, 5.0, -5.0],
[1.0, -1.0, 2.0, -2.0, 5.0, -5.0, 6.0, -6.0],
[2.0, -2.0, 3.0, -3.0, 6.0, -6.0, 7.0, -7.0],
[3.0, -3.0, 4.0, -4.0, 7.0, -7.0, 8.0, -8.0],
[4.0, -4.0, 0.0, 0.0, 8.0, -8.0, 0.0, 0.0],
],
[
[0.0, 5.0, 0.0, 9.0, 0.0, -5.0, 0.0, -9.0],
[5.0, 6.0, 9.0, 10.0, -5.0, -6.0, -9.0, -10.0],
[6.0, 7.0, 10.0, 11.0, -6.0, -7.0, -10.0, -11.0],
[7.0, 8.0, 11.0, 12.0, -7.0, -8.0, -11.0, -12.0],
[8.0, 0.0, 12.0, 0.0, -8.0, 0.0, -12.0, 0.0],
[0.0, 0.0, 5.0, -5.0, 0.0, 0.0, 9.0, -9.0],
[5.0, -5.0, 6.0, -6.0, 9.0, -9.0, 10.0, -10.0],
[6.0, -6.0, 7.0, -7.0, 10.0, -10.0, 11.0, -11.0],
[7.0, -7.0, 8.0, -8.0, 11.0, -11.0, 12.0, -12.0],
[8.0, -8.0, 0.0, 0.0, 12.0, -12.0, 0.0, 0.0],
],
[
[0.0, 9.0, 0.0, 13.0, 0.0, -9.0, 0.0, -13.0],
[9.0, 10.0, 13.0, 14.0, -9.0, -10.0, -13.0, -14.0],
[10.0, 11.0, 14.0, 15.0, -10.0, -11.0, -14.0, -15.0],
[11.0, 12.0, 15.0, 16.0, -11.0, -12.0, -15.0, -16.0],
[12.0, 0.0, 16.0, 0.0, -12.0, 0.0, -16.0, 0.0],
[0.0, 0.0, 9.0, -9.0, 0.0, 0.0, 13.0, -13.0],
[9.0, -9.0, 10.0, -10.0, 13.0, -13.0, 14.0, -14.0],
[10.0, -10.0, 11.0, -11.0, 14.0, -14.0, 15.0, -15.0],
[11.0, -11.0, 12.0, -12.0, 15.0, -15.0, 16.0, -16.0],
[12.0, -12.0, 0.0, 0.0, 16.0, -16.0, 0.0, 0.0],
],
[
[0.0, 13.0, 0.0, 0.0, 0.0, -13.0, 0.0, 0.0],
[13.0, 14.0, 0.0, 0.0, -13.0, -14.0, 0.0, 0.0],
[14.0, 15.0, 0.0, 0.0, -14.0, -15.0, 0.0, 0.0],
[15.0, 16.0, 0.0, 0.0, -15.0, -16.0, 0.0, 0.0],
[16.0, 0.0, 0.0, 0.0, -16.0, 0.0, 0.0, 0.0],
[0.0, 0.0, 13.0, -13.0, 0.0, 0.0, 0.0, 0.0],
[13.0, -13.0, 14.0, -14.0, 0.0, 0.0, 0.0, 0.0],
[14.0, -14.0, 15.0, -15.0, 0.0, 0.0, 0.0, 0.0],
[15.0, -15.0, 16.0, -16.0, 0.0, 0.0, 0.0, 0.0],
[16.0, -16.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
],
]
],
......@@ -281,7 +281,7 @@ def test_im2col_infer_shapes():
"outp", TensorProto.FLOAT, [1, ofm_dim, ofm_dim, k * k * ifm_ch]
)
abs_node = helper.make_node("Abs", inputs=["inp"], outputs=["abs"],)
abs_node = helper.make_node("Abs", inputs=["inp"], outputs=["abs"])
Im2Col_node = helper.make_node(
"Im2Col",
......@@ -293,7 +293,7 @@ def test_im2col_infer_shapes():
input_shape="(1,{},{},{})".format(ifm_dim, ifm_dim, ifm_ch),
)
abs1_node = helper.make_node("Abs", inputs=["im2col"], outputs=["outp"],)
abs1_node = helper.make_node("Abs", inputs=["im2col"], outputs=["outp"])
graph = helper.make_graph(
nodes=[abs_node, Im2Col_node, abs1_node],
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment