| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesIndexing.cpp | 16 TORCH_LIBRARY_IMPL(aten, FuncTorchBatched, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | LegacyBatchingRegistrations.cpp | 731 TORCH_LIBRARY_IMPL(_, FuncTorchBatched, m) { in TORCH_LIBRARY_IMPL() argument
735 TORCH_LIBRARY_IMPL(aten, FuncTorchBatched, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | BatchRulesDynamic.cpp | 66 TORCH_LIBRARY_IMPL(aten, FuncTorchBatched, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | BatchRulesUnaryOps.cpp | 64 TORCH_LIBRARY_IMPL(aten, FuncTorchBatched, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | BatchRulesActivation.cpp | 52 TORCH_LIBRARY_IMPL(aten, FuncTorchBatched, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | BatchRulesPooling.cpp | 56 TORCH_LIBRARY_IMPL(aten, FuncTorchBatched, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | BatchRulesLoss.cpp | 172 TORCH_LIBRARY_IMPL(aten, FuncTorchBatched, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | BatchRulesFactory.cpp | 225 TORCH_LIBRARY_IMPL(aten, FuncTorchBatched, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | BatchRulesModules.cpp | 348 TORCH_LIBRARY_IMPL(aten, FuncTorchBatched, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | BatchRulesReduceOps.cpp | 449 TORCH_LIBRARY_IMPL(aten, FuncTorchBatched, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | BatchRulesRandomness.cpp | 356 TORCH_LIBRARY_IMPL(aten, FuncTorchBatched, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | BatchRulesBinaryOps.cpp | 320 TORCH_LIBRARY_IMPL(aten, FuncTorchBatched, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | BatchRulesConvolution.cpp | 458 TORCH_LIBRARY_IMPL(aten, FuncTorchBatched, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | BatchRulesViews.cpp | 555 TORCH_LIBRARY_IMPL(aten, FuncTorchBatched, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | BatchRulesLinearAlgebra.cpp | 750 TORCH_LIBRARY_IMPL(aten, FuncTorchBatched, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | BatchRulesNorm.cpp | 864 TORCH_LIBRARY_IMPL(aten, FuncTorchBatched, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | BatchRulesScatterOps.cpp | 1230 TORCH_LIBRARY_IMPL(aten, FuncTorchBatched, m) { in TORCH_LIBRARY_IMPL() argument
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | DispatchKey.h | 371 FuncTorchBatched, // See Note [Out-of-tree vmap+grad prototype] enumerator
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | model.py | 118 FuncTorchBatched = auto() variable in DispatchKey
|