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

[CnvLower] add qnt annotations to newly created tensors

parent 002d5e78
No related branches found
No related tags found
No related merge requests found
......@@ -48,6 +48,8 @@ class LowerConvsToMatMul(Transformation):
graph_modified = True
cnv_input = n.input[0]
cnv_output = n.output[0]
idt = model.get_tensor_datatype(cnv_input)
odt = model.get_tensor_datatype(cnv_output)
# extract conv parameters
k = get_by_name(n.attribute, "kernel_shape").ints[-1]
pad = get_by_name(n.attribute, "pads").ints[-1]
......@@ -69,6 +71,7 @@ class LowerConvsToMatMul(Transformation):
)
graph.value_info.append(inp_trans_out)
inp_trans_out = inp_trans_out.name
model.set_tensor_datatype(inp_trans_out, idt)
im2col_out = helper.make_tensor_value_info(
model.make_new_valueinfo_name(),
......@@ -77,6 +80,7 @@ class LowerConvsToMatMul(Transformation):
)
graph.value_info.append(im2col_out)
im2col_out = im2col_out.name
model.set_tensor_datatype(im2col_out, idt)
matmul_out = helper.make_tensor_value_info(
model.make_new_valueinfo_name(),
......@@ -85,6 +89,7 @@ class LowerConvsToMatMul(Transformation):
)
graph.value_info.append(matmul_out)
matmul_out = matmul_out.name
model.set_tensor_datatype(matmul_out, odt)
# create new nodes
# NCHW -> NHWC
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment