Skip to content

Commit d280118

Browse files
committed
Set TOpPattern=kOpaque for scatter_nd
1 parent fe398bf commit d280118

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/relay/op/tensor/transform.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1158,7 +1158,7 @@ Given data with shape (Y_0, ..., Y_{K-1}, X_M, ..., X_{N-1}) and indices with sh
11581158
.add_argument("indices", "Tensor", "The indices tensor.")
11591159
.set_support_level(3)
11601160
.add_type_rel("ScatterND", ScatterNDRel)
1161-
.set_attr<TOpPattern>("TOpPattern", kInjective);
1161+
.set_attr<TOpPattern>("TOpPattern", kOpaque);
11621162

11631163
// Take
11641164
TVM_REGISTER_NODE_TYPE(TakeAttrs);

0 commit comments

Comments
 (0)