Home
last modified time | relevance | path

Searched refs:CompositeExplicitAutogradNonFunctional (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/torchgen/dest/
H A Dregister_dispatch_key.py99 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 DDispatchKeySet.cpp80 case DispatchKey::CompositeExplicitAutogradNonFunctional: in getRuntimeDispatchKeySet()
101 case DispatchKey::CompositeExplicitAutogradNonFunctional: in runtimeDispatchKeySetHas()
H A DDispatchKey.cpp206 case DispatchKey::CompositeExplicitAutogradNonFunctional: in toString()
379 c10::DispatchKey::CompositeExplicitAutogradNonFunctional}, in parseDispatchKey()
H A DDispatchKey.h479 CompositeExplicitAutogradNonFunctional, // registered at enumerator
485 EndOfAliasKeys = CompositeExplicitAutogradNonFunctional, //
/aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/
H A DOperatorEntry.cpp305 …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 Dgen.py331 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 Dgen_aoti_c_shim.py332 elif backend_indices[DispatchKey.CompositeExplicitAutogradNonFunctional].has_kernel(
337 DispatchKey.CompositeExplicitAutogradNonFunctional
H A Dmodel.py131 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 Dpython_dispatch.cpp704 DEF_ONE(CompositeExplicitAutogradNonFunctional) in initDispatchBindings()
/aosp_15_r20/external/pytorch/torch/
H A D_ops.py182 cand = DispatchKey.CompositeExplicitAutogradNonFunctional
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DREADME.md326 - `CompositeExplicitAutogradNonFunctional`:
/aosp_15_r20/external/pytorch/
H A Dbuckbuild.bzl290 "CompositeExplicitAutogradNonFunctional",