Skip to content
Snippets Groups Projects
Unverified Commit b6a7f60b authored by Yaman Umuroglu's avatar Yaman Umuroglu Committed by GitHub
Browse files

[Transform] set output tensor dt in InferChannelwiseLinearLayer

parent 37d052a0
No related branches found
No related tags found
No related merge requests found
......@@ -577,6 +577,7 @@ class InferChannelwiseLinearLayer(Transformation):
else:
odt = DataType.get_smallest_possible(idt.max() * idt.max())
model.set_initializer(ll_const, ll_cinit.reshape(ch))
model.set_tensor_datatype(ll_output, odt)
# create and insert node
new_node = helper.make_node(
"ChannelwiseOp_Batch",
......
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