Skip to content

Commit

Permalink
Wrap line
Browse files Browse the repository at this point in the history
  • Loading branch information
jroesch committed May 23, 2019
1 parent e67f10f commit 7c5d4d3
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion python/tvm/relay/frontend/onnx.py
Original file line number Diff line number Diff line change
Expand Up @@ -1018,7 +1018,10 @@ def from_onnx(self, graph, opset):
if len(array.shape) == 0 and array.dtype == 'int64':
array = _nd.array(array.asnumpy().astype('int32'))
self._params[node.output[0]] = array
self._nodes[node.output[0]] = new_var(node.output[0], shape=list(t_proto.dims), dtype=array.dtype)
self._nodes[node.output[0]] = new_var(
node.output[0],
shape=list(t_proto.dims),
dtype=array.dtype)
else:
if op_name == "ConstantFill":
fill_value = attr.get('value', 0.0)
Expand Down

0 comments on commit 7c5d4d3

Please sign in to comment.