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

[LabelSel] use UINT32 in dtype inference

parent af60e5de
No related branches found
No related tags found
No related merge requests found
......@@ -105,7 +105,9 @@ class LabelSelect_Batch(HLSCustomOp):
)
def infer_node_datatype(self, model):
model.set_tensor_datatype(self.onnx_node.output[0], DataType.INT64)
# currently set to uint32 to be compatible with hlslib
# enhancement: consider finding smallest power-of-two int for reduced output bandwidth
model.set_tensor_datatype(self.onnx_node.output[0], DataType.UINT32)
def verify_node(self):
info_messages = []
......
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