File tree Expand file tree Collapse file tree
onnxscript/function_libs/torch_lib/ops Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -3410,21 +3410,11 @@ def aten_embedding_bag_padding_idx(
34103410
34113411 if padding_idx is not None :
34123412 # Call the existing function for handling padding_idx
3413- < << << << HEAD
34143413 result , offset2bag , bag_size , max_indices = _aten_embedding_bag_1d_padding_idx_onnx (
34153414 weight ,
34163415 indices ,
34173416 offsets ,
34183417 mode ,
3419- == == == =
3420- result , offset2bag , bag_size , max_indices = _aten_embedding_bag_1d_padding_idx_onnx (
3421- weight ,
3422- indices ,
3423- offsets ,
3424- scale_grad_by_freq ,
3425- mode ,
3426- sparse ,
3427- >> >> >> > 6 b232d90 (added fixed function logic )
34283418 per_sample_weights ,
34293419 include_last_offset ,
34303420 padding_idx ,
@@ -3437,13 +3427,7 @@ def aten_embedding_bag_padding_idx(
34373427 weight ,
34383428 indices ,
34393429 offsets ,
3440- << << << < HEAD
3441- mode ,
3442- == == == =
3443- scale_grad_by_freq ,
34443430 mode ,
3445- sparse ,
3446- >> >> >> > 6 b232d90 (added fixed function logic )
34473431 per_sample_weights ,
34483432 include_last_offset ,
34493433 if padding_idx is not None :
You can’t perform that action at this time.
0 commit comments