We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see ourdocumentation.
There was an error while loading.Please reload this page.
1 parent7330b58 commitd00d823Copy full SHA for d00d823
onnx_array_api/graph_api/graph_builder.py
@@ -173,6 +173,9 @@ def _get_tensor_shape(
173
returntuple(att.floats)
174
ifatt.name=="value_ints":
175
return (len(att.ints),)
176
+ifatt.name=="value":
177
+t=onh.to_array(att.t)
178
+returnt.shape
179
raiseTypeError(
180
f"Unexpected or unsupported scenario type{type(proto)}:{proto}."
181
)
@@ -190,6 +193,9 @@ def _get_tensor_type(self, proto: Union[NodeProto, TensorProto]) -> int:
190
193
returnTensorProto.FLOAT
191
194
192
195
returnTensorProto.INT64
196
197
198
+returnoh.np_dtype_to_tensor_dtype(t.dtype)
199
raiseValueError(f"Unexpected type or value{type(proto)}:{proto}.")
200
201
defis_constant(self,name:str)->bool: