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",
|