Searched defs:ind_inv (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/testing/_internal/ |
| D | autograd_function_db.py | 296 def forward(x, ind, ind_inv, dim): argument
316 def vmap(info, in_dims, x, ind, ind_inv, dim): argument
345 def forward(x, ind, ind_inv, dim): argument
|
| D | custom_op_db.py | 162 def _(x, ind, ind_inv, dim): argument
181 def numpy_take_vmap(info, in_dims, x, ind, ind_inv, dim): argument
|