Searched refs:handle_from_tensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/functorch/csrc/dim/ |
| H A D | dim.cpp | 1113 mpy::handle handle_from_tensor(Arena& A, TensorRef t) { in handle_from_tensor() function
1260 flat_args[i] = handle_from_tensor(A, std::move(ml)); in run_torch_function()
1291 flat_args[i] = handle_from_tensor(A, batched); in run_torch_function()
2101 auto self_hdl = handle_from_tensor(A, iinfo.self); in invoke_getitem()
2587 …flat_inputs[i] = handle_from_tensor(A, _match_levels(A, t, tensor_inputs[i].levels, index_levels)); in getsetitem_flat()
2639 rhs = handle_from_tensor(A, rhs_matched); in __setitem__()
2641 self = handle_from_tensor(A, iinfo.self); in __setitem__()
2972 patched_args[0] = handle_from_tensor(A, info.batchedtensor); in patched_dim_method()
3022 patched_args[0] = handle_from_tensor(A, info.tensor); in patched_dim_method()
|