Searched defs:FuncTorchVmapMode (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | VmapModeRegistrations.cpp | 27 TORCH_LIBRARY_IMPL(_, FuncTorchVmapMode, m) { in TORCH_LIBRARY_IMPL() argument
48 TORCH_LIBRARY_IMPL(aten, FuncTorchVmapMode, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | BatchRulesDecompositions.cpp | 21 TORCH_LIBRARY_IMPL(aten, FuncTorchVmapMode, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | BatchRulesRandomness.cpp | 367 TORCH_LIBRARY_IMPL(aten, FuncTorchVmapMode, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | BatchRulesBinaryOps.cpp | 310 TORCH_LIBRARY_IMPL(aten, FuncTorchVmapMode, m) { in TORCH_LIBRARY_IMPL() argument
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | DispatchKey.h | 376 FuncTorchVmapMode, // See Note [Out-of-tree vmap+grad prototype] enumerator
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | model.py | 120 FuncTorchVmapMode = auto() variable in DispatchKey
|