| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | VariableFallbackKernel.cpp | 63 TORCH_LIBRARY_IMPL(_, AutogradOther, m) { in TORCH_LIBRARY_IMPL() function
67 TORCH_LIBRARY_IMPL(_, AutogradCPU, m) { in TORCH_LIBRARY_IMPL() function
71 TORCH_LIBRARY_IMPL(_, AutogradXPU, m) { in TORCH_LIBRARY_IMPL() function
75 TORCH_LIBRARY_IMPL(_, AutogradCUDA, m) { in TORCH_LIBRARY_IMPL() function
79 TORCH_LIBRARY_IMPL(_, AutogradXLA, m) { in TORCH_LIBRARY_IMPL() function
83 TORCH_LIBRARY_IMPL(_, AutogradLazy, m) { in TORCH_LIBRARY_IMPL() function
87 TORCH_LIBRARY_IMPL(_, AutogradMPS, m) { in TORCH_LIBRARY_IMPL() function
91 TORCH_LIBRARY_IMPL(_, AutogradMeta, m) { in TORCH_LIBRARY_IMPL() function
96 TORCH_LIBRARY_IMPL(_, ADInplaceOrView, m) { in TORCH_LIBRARY_IMPL() function
100 TORCH_LIBRARY_IMPL(_, AutogradHPU, m) { in TORCH_LIBRARY_IMPL() function
|
| H A D | PythonFallbackKernel.cpp | 153 TORCH_LIBRARY_IMPL(_, Python, m) { in TORCH_LIBRARY_IMPL() function
157 TORCH_LIBRARY_IMPL(_, PythonDispatcher, m) { in TORCH_LIBRARY_IMPL() function
161 TORCH_LIBRARY_IMPL(_, PythonTLSSnapshot, m) { in TORCH_LIBRARY_IMPL() function
165 TORCH_LIBRARY_IMPL(_, PreDispatch, m) { in TORCH_LIBRARY_IMPL() function
|
| H A D | NamedRegistrations.cpp | 7 TORCH_LIBRARY_IMPL(_, Named, m) { in TORCH_LIBRARY_IMPL() function
11 TORCH_LIBRARY_IMPL(aten, Named, m) { in TORCH_LIBRARY_IMPL() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | autocast_mode.cpp | 165 TORCH_LIBRARY_IMPL(_, Autocast, m) { in TORCH_LIBRARY_IMPL() function
169 TORCH_LIBRARY_IMPL(aten, Autocast, m) { in TORCH_LIBRARY_IMPL() function
209 TORCH_LIBRARY_IMPL(_, AutocastMPS, m) { in TORCH_LIBRARY_IMPL() function
213 TORCH_LIBRARY_IMPL(aten, AutocastMPS, m) { in TORCH_LIBRARY_IMPL() function
321 TORCH_LIBRARY_IMPL(_, AutocastCPU, m) { in TORCH_LIBRARY_IMPL() function
326 TORCH_LIBRARY_IMPL(aten, AutocastCPU, m) { in TORCH_LIBRARY_IMPL() function
463 TORCH_LIBRARY_IMPL(_, AutocastXPU, m) { in TORCH_LIBRARY_IMPL() function
467 TORCH_LIBRARY_IMPL(aten, AutocastXPU, m) { in TORCH_LIBRARY_IMPL() function
|
| H A D | VmapModeRegistrations.cpp | 33 TORCH_LIBRARY_IMPL(_, VmapMode, m) { in TORCH_LIBRARY_IMPL() function
37 TORCH_LIBRARY_IMPL(aten, VmapMode, m) { in TORCH_LIBRARY_IMPL() function
|
| H A D | ConjugateFallback.cpp | 17 TORCH_LIBRARY_IMPL(_, Conjugate, m) { in TORCH_LIBRARY_IMPL() function
21 TORCH_LIBRARY_IMPL(aten, Conjugate, m) { in TORCH_LIBRARY_IMPL() function
|
| H A D | ZeroTensorFallback.cpp | 86 TORCH_LIBRARY_IMPL(_, ZeroTensor, m) { in TORCH_LIBRARY_IMPL() function
90 TORCH_LIBRARY_IMPL(aten, ZeroTensor, m) { in TORCH_LIBRARY_IMPL() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Register.cpp | 269 TORCH_LIBRARY_IMPL(vulkan_prepack, CPU, m) { in TORCH_LIBRARY_IMPL() function
299 TORCH_LIBRARY_IMPL(vulkan_prepack, QuantizedCPU, m) { in TORCH_LIBRARY_IMPL() function
308 TORCH_LIBRARY_IMPL(vulkan_prepack, Vulkan, m) { in TORCH_LIBRARY_IMPL() function
367 TORCH_LIBRARY_IMPL(vulkan_quantized, Vulkan, m) { in TORCH_LIBRARY_IMPL() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qlinear_prepack.cpp | 675 TORCH_LIBRARY_IMPL(quantized, QuantizedCPU, m) { in TORCH_LIBRARY_IMPL() function
681 TORCH_LIBRARY_IMPL(quantized, CPU, m) { in TORCH_LIBRARY_IMPL() function
687 TORCH_LIBRARY_IMPL(_quantized, QuantizedCPU, m) { in TORCH_LIBRARY_IMPL() function
692 TORCH_LIBRARY_IMPL(_quantized, CPU, m) { in TORCH_LIBRARY_IMPL() function
705 TORCH_LIBRARY_IMPL(_quantized, Meta, m) { in TORCH_LIBRARY_IMPL() function
715 TORCH_LIBRARY_IMPL(onednn, CPU, m) { in TORCH_LIBRARY_IMPL() function
|
| H A D | qlinear_dynamic.cpp | 791 TORCH_LIBRARY_IMPL(quantized, CPU, m) { in TORCH_LIBRARY_IMPL() function
810 TORCH_LIBRARY_IMPL(quantized, Meta, m) { in TORCH_LIBRARY_IMPL() function
816 TORCH_LIBRARY_IMPL(_quantized, CPU, m) { in TORCH_LIBRARY_IMPL() function
829 TORCH_LIBRARY_IMPL(_quantized, Meta, m) { in TORCH_LIBRARY_IMPL() function
|
| H A D | qembeddingbag_unpack.cpp | 269 TORCH_LIBRARY_IMPL(quantized, CPU, m) { in TORCH_LIBRARY_IMPL() function
281 TORCH_LIBRARY_IMPL(quantized, CatchAll, m) { in TORCH_LIBRARY_IMPL() function
289 TORCH_LIBRARY_IMPL(quantized, Meta, m) { in TORCH_LIBRARY_IMPL() function
|
| H A D | qembeddingbag_prepack.cpp | 540 TORCH_LIBRARY_IMPL(quantized, CPU, m) { in TORCH_LIBRARY_IMPL() function
552 TORCH_LIBRARY_IMPL(quantized, QuantizedCPU, m) { in TORCH_LIBRARY_IMPL() function
558 TORCH_LIBRARY_IMPL(quantized, Meta, m) { in TORCH_LIBRARY_IMPL() function
|
| H A D | qconv_prepack.cpp | 828 TORCH_LIBRARY_IMPL(quantized, QuantizedCPU, m) { in TORCH_LIBRARY_IMPL() function
841 TORCH_LIBRARY_IMPL(_quantized, QuantizedCPU, m) { in TORCH_LIBRARY_IMPL() function
851 TORCH_LIBRARY_IMPL(onednn, CPU, m) { in TORCH_LIBRARY_IMPL() function
|
| H A D | qlinear.cpp | 1321 TORCH_LIBRARY_IMPL(quantized, QuantizedCPU, m) { in TORCH_LIBRARY_IMPL() function
1329 TORCH_LIBRARY_IMPL(_quantized, QuantizedCPU, m) { in TORCH_LIBRARY_IMPL() function
1334 TORCH_LIBRARY_IMPL(quantized, CPU, m) { in TORCH_LIBRARY_IMPL() function
1339 TORCH_LIBRARY_IMPL(onednn, MkldnnCPU, m) { in TORCH_LIBRARY_IMPL() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Linear.cpp | 419 TORCH_LIBRARY_IMPL(mkl, CPU, m) { in TORCH_LIBRARY_IMPL() function
423 TORCH_LIBRARY_IMPL(mkl, MkldnnCPU, m) { in TORCH_LIBRARY_IMPL() function
429 TORCH_LIBRARY_IMPL(mkldnn, CPU, m) { in TORCH_LIBRARY_IMPL() function
438 TORCH_LIBRARY_IMPL(mkldnn, MkldnnCPU, m) { in TORCH_LIBRARY_IMPL() function
|
| H A D | MKLDNNConversions.cpp | 541 TORCH_LIBRARY_IMPL(mkldnn, CPU, m) { in TORCH_LIBRARY_IMPL() function
556 TORCH_LIBRARY_IMPL(mkldnn, MkldnnCPU, m) { in TORCH_LIBRARY_IMPL() function
632 TORCH_LIBRARY_IMPL(mkl, CPU, m) { in TORCH_LIBRARY_IMPL() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | LegacyBatchingRegistrations.cpp | 731 TORCH_LIBRARY_IMPL(_, FuncTorchBatched, m) { in TORCH_LIBRARY_IMPL() function
735 TORCH_LIBRARY_IMPL(aten, FuncTorchBatched, m) { in TORCH_LIBRARY_IMPL() function
758 TORCH_LIBRARY_IMPL(_, BatchedNestedTensor, m) { in TORCH_LIBRARY_IMPL() function
763 TORCH_LIBRARY_IMPL(aten, BatchedNestedTensor, m) { in TORCH_LIBRARY_IMPL() function
|
| H A D | BatchRulesDecompositions.cpp | 21 TORCH_LIBRARY_IMPL(aten, FuncTorchVmapMode, m) { in TORCH_LIBRARY_IMPL() function
35 TORCH_LIBRARY_IMPL(aten, FuncTorchBatchedDecomposition, m) { in TORCH_LIBRARY_IMPL() function
|
| H A D | VmapModeRegistrations.cpp | 27 TORCH_LIBRARY_IMPL(_, FuncTorchVmapMode, m) { in TORCH_LIBRARY_IMPL() function
48 TORCH_LIBRARY_IMPL(aten, FuncTorchVmapMode, m) { in TORCH_LIBRARY_IMPL() function
|
| H A D | DynamicLayer.cpp | 493 TORCH_LIBRARY_IMPL(_, FuncTorchDynamicLayerFrontMode, m) { in TORCH_LIBRARY_IMPL() function
497 TORCH_LIBRARY_IMPL(_, FuncTorchDynamicLayerBackMode, m) { in TORCH_LIBRARY_IMPL() function
505 TORCH_LIBRARY_IMPL(aten, FuncTorchDynamicLayerBackMode, m) { in TORCH_LIBRARY_IMPL() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | NegateFallback.cpp | 18 TORCH_LIBRARY_IMPL(_, Negative, m) { in TORCH_LIBRARY_IMPL() function
22 TORCH_LIBRARY_IMPL(aten, Negative, m) { in TORCH_LIBRARY_IMPL() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/ |
| H A D | qlinear_unpack.cpp | 67 TORCH_LIBRARY_IMPL(quantized, CPU, m) { in TORCH_LIBRARY_IMPL() function
72 TORCH_LIBRARY_IMPL(quantized, CatchAll, m) { in TORCH_LIBRARY_IMPL() function
|
| /aosp_15_r20/external/pytorch/test/cpp/jit/ |
| H A D | test_custom_class_registrations.cpp | 702 TORCH_LIBRARY_IMPL(_TorchScriptTesting, CPU, m) { in TORCH_LIBRARY_IMPL() function
711 TORCH_LIBRARY_IMPL(_TorchScriptTesting, Meta, m) { in TORCH_LIBRARY_IMPL() function
717 TORCH_LIBRARY_IMPL(_TorchScriptTesting, CompositeImplicitAutograd, m) { in TORCH_LIBRARY_IMPL() function
723 TORCH_LIBRARY_IMPL(_TorchScriptTesting, BackendSelect, m) { in TORCH_LIBRARY_IMPL() function
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | TraceTypeManual.cpp | 131 TORCH_LIBRARY_IMPL(aten, Tracer, m) { in TORCH_LIBRARY_IMPL() function
294 TORCH_LIBRARY_IMPL(_, Tracer, m) { in TORCH_LIBRARY_IMPL() function
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | Ops.cpp | 526 TORCH_LIBRARY_IMPL(c10d, SparseCPU, m) { in TORCH_LIBRARY_IMPL() function
530 TORCH_LIBRARY_IMPL(c10d, SparseCUDA, m) { in TORCH_LIBRARY_IMPL() function
|