Skip to content

Commit ca6db1a

Browse files
committed
fixed style issue
1 parent 9ba946e commit ca6db1a

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

  • onnxscript/function_libs/torch_lib/ops

onnxscript/function_libs/torch_lib/ops/core.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2964,7 +2964,7 @@ def _aten_embedding_bag_onnx(
29642964
max_indices = op.Cast(op.Expand(0, op.Shape(offsets)), to=INT64.dtype)
29652965
else:
29662966
bag_size = op.Cast(op.Expand(0, op.Shape(offsets) - 1), to=INT64.dtype)
2967-
max_indices = op.Cast(op.Expand(0, op.Shape(offsets) -1), to=INT64.dtype)
2967+
max_indices = op.Cast(op.Expand(0, op.Shape(offsets) - 1), to=INT64.dtype)
29682968
elif mode == 1: # mean
29692969
offset2bag = op.Cast(op.Expand(0, op.Shape(indices, start=0, end=1)), to=INT64.dtype)
29702970
bag_size = op.Cast(op.Expand(0, op.Shape(offsets) - 1), to=INT64.dtype)

0 commit comments

Comments
 (0)