Skip to content
Snippets Groups Projects
Commit 153c2d4e authored by mmrahorovic's avatar mmrahorovic
Browse files

[custom op]: update tensor datatype for consistency

parent 5615d8d3
No related branches found
No related tags found
No related merge requests found
......@@ -211,6 +211,8 @@ class Thresholding_Batch(HLSCustomOp):
threshold_tensor
).all(), "Thresholds can't be expressed with type %s" % str(tdt)
self.set_nodeattr("weightDataType", tdt.name)
# Update QONNX DataType of tensor for consistency
model.set_tensor_datatype(self.onnx_node.input[1], tdt)
return DataType[self.get_nodeattr("weightDataType")]
def get_instream_width(self, ind=0):
......
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