Skip to content

Commit 8b2f4b7

Browse files
committed
const dtype is needed.
for example it's needed when inserting cast for target rs6
1 parent 71747f3 commit 8b2f4b7

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tf2onnx/graph.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -414,8 +414,9 @@ def make_const(self, name, np_val, skip_conversion=False, raw=True):
414414
else:
415415
onnx_tensor = helper.make_tensor(name, utils.map_numpy_to_onnx_dtype(np_val.dtype),
416416
np_val.shape, np_val, raw=False)
417+
dtype = onnx_tensor.data_type
417418
node = self.make_node("Const", [], outputs=[name], name=name, attr={"value": onnx_tensor},
418-
skip_conversion=skip_conversion)
419+
skip_conversion=skip_conversion, dtypes=[dtype])
419420
self.set_shape(name, np_val.shape)
420421
self.set_dtype(name, utils.map_numpy_to_onnx_dtype(np_val.dtype))
421422
return node

0 commit comments

Comments
 (0)