Searched refs:CompositeImplicitAutogradNestedTensor (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | DispatchKeySet.cpp | 76 case DispatchKey::CompositeImplicitAutogradNestedTensor: in getRuntimeDispatchKeySet()
95 case DispatchKey::CompositeImplicitAutogradNestedTensor: in runtimeDispatchKeySetHas()
|
| H A D | DispatchKey.cpp | 202 case DispatchKey::CompositeImplicitAutogradNestedTensor: in toString()
375 c10::DispatchKey::CompositeImplicitAutogradNestedTensor}, in parseDispatchKey()
|
| H A D | DispatchKey.h | 474 CompositeImplicitAutogradNestedTensor, // registered at enumerator
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | model.py | 129 CompositeImplicitAutogradNestedTensor = auto() variable in DispatchKey
287 DispatchKey.CompositeImplicitAutogradNestedTensor,
310 DispatchKey.CompositeImplicitAutogradNestedTensor,
814 or d == DispatchKey.CompositeImplicitAutogradNestedTensor
824 DispatchKey.CompositeImplicitAutogradNestedTensor
885 != {DispatchKey.CompositeImplicitAutogradNestedTensor}
889 DispatchKey.CompositeImplicitAutogradNestedTensor,
897 DispatchKey.CompositeImplicitAutogradNestedTensor in dispatch
|
| H A D | gen.py | 329 DispatchKey.CompositeImplicitAutogradNestedTensor,
351 return DispatchKey.CompositeImplicitAutogradNestedTensor
1314 DispatchKey.CompositeImplicitAutogradNestedTensor,
2275 dispatch_key != DispatchKey.CompositeImplicitAutogradNestedTensor
2897 DispatchKey.CompositeImplicitAutogradNestedTensor,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/ |
| H A D | OperatorEntry.cpp | 339 …Undefined && isIncludedInAlias(dispatch_key, DispatchKey::CompositeImplicitAutogradNestedTensor)) { in registerKernel()
340 …ested_registration = getKernelForDispatchKey(DispatchKey::CompositeImplicitAutogradNestedTensor)) { in registerKernel()
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | python_dispatch.cpp | 706 DEF_ONE(CompositeImplicitAutogradNestedTensor) in initDispatchBindings()
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _ops.py | 197 cand = DispatchKey.CompositeImplicitAutogradNestedTensor
|
| /aosp_15_r20/external/pytorch/ |
| H A D | buckbuild.bzl | 292 "CompositeImplicitAutogradNestedTensor",
|