We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 700bb1a commit fde4802Copy full SHA for fde4802
onnxscript/function_libs/torch_lib/ops/core.py
@@ -7592,7 +7592,7 @@ def aten_scalar_tensor_sym_number(
7592
return common_ops.cast_to(s, dtype=dtype)
7593
7594
7595
-@torch_op("aten::scatter.value", trace_only=True)
+@torch_op(("aten::scatter.value", "aten::scatter.src"), trace_only=True)
7596
def aten_scatter(
7597
self: TReal,
7598
dim: int, # we have to use int here because ScatterElements() will use this attribute
0 commit comments