Searched refs:run_torch_function (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/functorch/csrc/dim/ |
| H A D | dim.cpp | 625 mpy::object run_torch_function(Arena &A, mpy::handle orig, mpy::vector_args args, bool is_pointwise…
709 auto t = Tensor::wrap(run_torch_function(A, delayed_->orig, delayed_->args, true)); in tensor()
1223 mpy::object run_torch_function(Arena &A, mpy::handle orig, mpy::vector_args args, bool is_pointwise… in run_torch_function() function
1325 return run_torch_function(A, orig, args, is_pointwise); in __torch_function__()
1338 return run_torch_function(A, orig, args, is_pointwise); in __torch_function__()
|