Searched refs:CompositeExplicitAutogradNonFunctional (Results 1 – 12 of 12) sorted by relevance
| /aosp_15_r20/external/pytorch/torchgen/dest/ |
| H A D | register_dispatch_key.py | 99 DispatchKey.CompositeExplicitAutogradNonFunctional,
152 if backend_index.dispatch_key == DispatchKey.CompositeExplicitAutogradNonFunctional:
372 == DispatchKey.CompositeExplicitAutogradNonFunctional
616 DispatchKey.CompositeExplicitAutogradNonFunctional,
648 DispatchKey.CompositeExplicitAutogradNonFunctional,
719 == DispatchKey.CompositeExplicitAutogradNonFunctional
768 == DispatchKey.CompositeExplicitAutogradNonFunctional
829 is DispatchKey.CompositeExplicitAutogradNonFunctional
922 == DispatchKey.CompositeExplicitAutogradNonFunctional
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | DispatchKeySet.cpp | 80 case DispatchKey::CompositeExplicitAutogradNonFunctional: in getRuntimeDispatchKeySet()
101 case DispatchKey::CompositeExplicitAutogradNonFunctional: in runtimeDispatchKeySetHas()
|
| H A D | DispatchKey.cpp | 206 case DispatchKey::CompositeExplicitAutogradNonFunctional: in toString()
379 c10::DispatchKey::CompositeExplicitAutogradNonFunctional}, in parseDispatchKey()
|
| H A D | DispatchKey.h | 479 CompositeExplicitAutogradNonFunctional, // registered at enumerator
485 EndOfAliasKeys = CompositeExplicitAutogradNonFunctional, //
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/ |
| H A D | OperatorEntry.cpp | 305 …ndefined || isIncludedInAlias(dispatch_key, DispatchKey::CompositeExplicitAutogradNonFunctional)) { in registerKernel()
306 …kend_registration = getKernelForDispatchKey(DispatchKey::CompositeExplicitAutogradNonFunctional)) { in registerKernel()
418 || dispatch_key == DispatchKey::CompositeExplicitAutogradNonFunctional) { in registerKernel()
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | gen.py | 331 DispatchKey.CompositeExplicitAutogradNonFunctional,
347 return DispatchKey.CompositeExplicitAutogradNonFunctional
2231 DispatchKey.CompositeExplicitAutogradNonFunctional,
2240 == DispatchKey.CompositeExplicitAutogradNonFunctional
2254 if dispatch_key == DispatchKey.CompositeExplicitAutogradNonFunctional:
2680 DispatchKey.CompositeExplicitAutogradNonFunctional
2899 DispatchKey.CompositeExplicitAutogradNonFunctional,
|
| H A D | gen_aoti_c_shim.py | 332 elif backend_indices[DispatchKey.CompositeExplicitAutogradNonFunctional].has_kernel(
337 DispatchKey.CompositeExplicitAutogradNonFunctional
|
| H A D | model.py | 131 CompositeExplicitAutogradNonFunctional = auto() variable in DispatchKey
289 DispatchKey.CompositeExplicitAutogradNonFunctional,
308 DispatchKey.CompositeExplicitAutogradNonFunctional,
812 or d == DispatchKey.CompositeExplicitAutogradNonFunctional
820 DispatchKey.CompositeExplicitAutogradNonFunctional
903 DispatchKey.CompositeExplicitAutogradNonFunctional in dispatch
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | python_dispatch.cpp | 704 DEF_ONE(CompositeExplicitAutogradNonFunctional) in initDispatchBindings()
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _ops.py | 182 cand = DispatchKey.CompositeExplicitAutogradNonFunctional
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | README.md | 326 - `CompositeExplicitAutogradNonFunctional`:
|
| /aosp_15_r20/external/pytorch/ |
| H A D | buckbuild.bzl | 290 "CompositeExplicitAutogradNonFunctional",
|