Skip to content

Commit

Permalink
update do
Browse files Browse the repository at this point in the history
  • Loading branch information
masahi committed May 20, 2021
1 parent c8b5a5c commit ffead4b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/relay/op/tensor/transform.cc
Original file line number Diff line number Diff line change
Expand Up @@ -3394,7 +3394,7 @@ Let B denote batch_dims, and data, indices shape be (X_0, X_1, ..., X_{N-1}),
and it must be the case that X_0, ... X_{B-1} == Y_0, ... Y_{B-1}.
The output will have shape
(Y_0, ..., Y_{K-1}, X_{M+B}, ..., X_{N-1}), where M + B <= N. If M + B == N,
(Y_0, ..., Y_{B-1}, ..., Y_{K-1}, X_{M+B}, ..., X_{N-1}), where M + B <= N. If M + B == N,
output shape will simply be (Y_0, ..., Y_{K-1}).
)code" TVM_ADD_FILELINE)
.set_num_inputs(2)
Expand Down

0 comments on commit ffead4b

Please sign in to comment.