Searched defs:CompositeExplicitAutograd (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/executorch/examples/portable/custom_ops/ |
| H A D | custom_ops_2.cpp | 31 TORCH_LIBRARY_IMPL(my_ops, CompositeExplicitAutograd, m) { in TORCH_LIBRARY_IMPL() argument
|
| /aosp_15_r20/external/executorch/extension/llm/custom_ops/ |
| H A D | op_fast_hadamard_transform_aten.cpp | 35 TORCH_LIBRARY_IMPL(llama, CompositeExplicitAutograd, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | op_tile_crop_aot.cpp | 45 TORCH_LIBRARY_IMPL(preprocess, CompositeExplicitAutograd, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | op_sdpa_aot.cpp | 180 TORCH_LIBRARY_IMPL(llama, CompositeExplicitAutograd, m) { in TORCH_LIBRARY_IMPL() argument
|
| /aosp_15_r20/external/executorch/kernels/quantized/test/ |
| H A D | quantized_ops_aot_register.cpp | 131 TORCH_LIBRARY_IMPL(et_quant_test, CompositeExplicitAutograd, m) { in TORCH_LIBRARY_IMPL() argument
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | DispatchKey.h | 476 CompositeExplicitAutograd, // registered at enumerator
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | custom_ops.cpp | 347 TORCH_LIBRARY_IMPL(aoti_custom_ops, CompositeExplicitAutograd, m) { in TORCH_LIBRARY_IMPL() argument
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | model.py | 130 CompositeExplicitAutograd = auto() variable in DispatchKey
|