|
|
@@ -4782,6 +4782,18 @@
|
|
|
{ "name": "output", "type": "Tensor" }
|
|
|
]
|
|
|
},
|
|
|
+ {
|
|
|
+ "name": "torch.scatter:src",
|
|
|
+ "inputs": [
|
|
|
+ { "name": "self", "type": "Tensor" },
|
|
|
+ { "name": "dim", "type": "int64" },
|
|
|
+ { "name": "index", "type": "Tensor" },
|
|
|
+ { "name": "src", "type": "Tensor" }
|
|
|
+ ],
|
|
|
+ "outputs": [
|
|
|
+ { "name": "output", "type": "Tensor" }
|
|
|
+ ]
|
|
|
+ },
|
|
|
{
|
|
|
"name": "torch.scatter_:src",
|
|
|
"inputs": [
|
|
|
@@ -4794,6 +4806,18 @@
|
|
|
{ "name": "output", "type": "Tensor" }
|
|
|
]
|
|
|
},
|
|
|
+ {
|
|
|
+ "name": "torch.scatter:value",
|
|
|
+ "inputs": [
|
|
|
+ { "name": "self", "type": "Tensor" },
|
|
|
+ { "name": "dim", "type": "int64" },
|
|
|
+ { "name": "index", "type": "Tensor" },
|
|
|
+ { "name": "src", "type": "Scalar" }
|
|
|
+ ],
|
|
|
+ "outputs": [
|
|
|
+ { "name": "output", "type": "Tensor" }
|
|
|
+ ]
|
|
|
+ },
|
|
|
{
|
|
|
"name": "torch.scatter_:value",
|
|
|
"inputs": [
|
|
|
@@ -4806,6 +4830,58 @@
|
|
|
{ "name": "output", "type": "Tensor" }
|
|
|
]
|
|
|
},
|
|
|
+ {
|
|
|
+ "name": "torch.scatter:reduce",
|
|
|
+ "inputs": [
|
|
|
+ { "name": "self", "type": "Tensor" },
|
|
|
+ { "name": "dim", "type": "int64" },
|
|
|
+ { "name": "index", "type": "Tensor" },
|
|
|
+ { "name": "src", "type": "Tensor" },
|
|
|
+ { "name": "reduce", "type": "string" }
|
|
|
+ ],
|
|
|
+ "outputs": [
|
|
|
+ { "name": "output", "type": "Tensor" }
|
|
|
+ ]
|
|
|
+ },
|
|
|
+ {
|
|
|
+ "name": "torch.scatter_:reduce",
|
|
|
+ "inputs": [
|
|
|
+ { "name": "self", "type": "Tensor" },
|
|
|
+ { "name": "dim", "type": "int64" },
|
|
|
+ { "name": "index", "type": "Tensor" },
|
|
|
+ { "name": "src", "type": "Tensor" },
|
|
|
+ { "name": "reduce", "type": "string" }
|
|
|
+ ],
|
|
|
+ "outputs": [
|
|
|
+ { "name": "output", "type": "Tensor" }
|
|
|
+ ]
|
|
|
+ },
|
|
|
+ {
|
|
|
+ "name": "torch.scatter:value_reduce",
|
|
|
+ "inputs": [
|
|
|
+ { "name": "self", "type": "Tensor" },
|
|
|
+ { "name": "dim", "type": "int64" },
|
|
|
+ { "name": "index", "type": "Tensor" },
|
|
|
+ { "name": "src", "type": "Scalar" },
|
|
|
+ { "name": "reduce", "type": "string" }
|
|
|
+ ],
|
|
|
+ "outputs": [
|
|
|
+ { "name": "output", "type": "Tensor" }
|
|
|
+ ]
|
|
|
+ },
|
|
|
+ {
|
|
|
+ "name": "torch.scatter_:value_reduce",
|
|
|
+ "inputs": [
|
|
|
+ { "name": "self", "type": "Tensor" },
|
|
|
+ { "name": "dim", "type": "int64" },
|
|
|
+ { "name": "index", "type": "Tensor" },
|
|
|
+ { "name": "src", "type": "Scalar" },
|
|
|
+ { "name": "reduce", "type": "string" }
|
|
|
+ ],
|
|
|
+ "outputs": [
|
|
|
+ { "name": "output", "type": "Tensor" }
|
|
|
+ ]
|
|
|
+ },
|
|
|
{
|
|
|
"name": "torch.scatter_add_",
|
|
|
"inputs": [
|