Open
Description
-
Remove assignment and return the last expression
@script() def aten_split(self, split_size): val_return_val = op.SplitToSequence(val_self, val_split_size, axis=0) return return_val
should become
@script() def aten_split(self, split_size): return op.SplitToSequence(val_self, val_split_size, axis=0)
-
Transform if graphs into branches
-
Normalize invalid node names
-
Handle function domain names: Currently the domain name is appended onto the function name, but that is not a valid name
-
Use
value_ints
etc. to replacemake_tensor
callsval_414 = op.Constant( value=onnx.helper.make_tensor( name="value", data_type=onnx.TensorProto.INT64, dims=[2], vals=[-1, 8] ) )