Searched full:compositeimplicitautogradnestedtensor (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | DispatchKey.cpp | 202 case DispatchKey::CompositeImplicitAutogradNestedTensor: in toString()
203 return "CompositeImplicitAutogradNestedTensor"; in toString()
374 {"CompositeImplicitAutogradNestedTensor", in parseDispatchKey()
375 c10::DispatchKey::CompositeImplicitAutogradNestedTensor}, in parseDispatchKey()
|
| H A D | DispatchKeySet.cpp | 76 case DispatchKey::CompositeImplicitAutogradNestedTensor: in getRuntimeDispatchKeySet()
95 case DispatchKey::CompositeImplicitAutogradNestedTensor: in runtimeDispatchKeySetHas()
|
| H A D | DispatchKey.h | 472 // Note: The alias keyset for CompositeImplicitAutogradNestedTensor is
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
2708 … " both have CompositeImplicitAutogradNestedTensor kernels, or both not have composite kernels."
|
| H A D | gen.py | 329 DispatchKey.CompositeImplicitAutogradNestedTensor,
351 return DispatchKey.CompositeImplicitAutogradNestedTensor
457 …return f"return {ns}::{DispatchKey.CompositeImplicitAutogradNestedTensor.lower()}::{name}({exprs})…
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 | 332 // If the dispatch key is included in CompositeImplicitAutogradNestedTensor, in registerKernel()
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",
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | native_functions.yaml | 4740 CompositeExplicitAutograd, CompositeImplicitAutogradNestedTensor: randn_like
4881 CompositeImplicitAutogradNestedTensor: reshape_nested_symint
4912 CompositeImplicitAutogradNestedTensor: reshape_as_nested
6521 CompositeExplicitAutograd, CompositeImplicitAutogradNestedTensor: zeros_like
|