We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a441ffc commit 417e7d9Copy full SHA for 417e7d9
onnx_diagnostic/ort_session.py
@@ -241,7 +241,7 @@ def _ortvalues_to_numpy_tensor(
241
el_type = ortvalues[i].element_type()
242
if el_type < onnx.TensorProto.BFLOAT16:
243
try:
244
- a = np.from_dlpack(ortvalues[i].to_dlpack())
+ a = np.from_dlpack(ortvalues[i])
245
except RuntimeError as e:
246
assert "ORT only supports contiguous tensor for now." in str(e), (
247
f"As it says, non-contiguous OrtValue are not supported "
0 commit comments