| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | Backend.cpp | 9 C10_LOG_API_USAGE_ONCE("c10d.backend"); in Backend()
15 C10_LOG_API_USAGE_ONCE(fmt::format("c10d.backend_{}", getBackendName())); in init()
|
| H A D | ProcessGroup.cpp | 130 C10_LOG_API_USAGE_ONCE("c10d.process_group"); in ProcessGroup()
139 C10_LOG_API_USAGE_ONCE( in init()
|
| H A D | reducer.cpp | 123 C10_LOG_API_USAGE_ONCE("torch.distributed.ddp.reducer"); in Reducer()
|
| H A D | init.cpp | 434 C10_LOG_API_USAGE_ONCE("c10d.python.import"); in c10d_init()
|
| H A D | ProcessGroupGloo.cpp | 2945 C10_LOG_API_USAGE_ONCE("torch.distributed.monitored_barrier"); in monitoredBarrier()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/python/ |
| H A D | python_tracer.cpp | 83 C10_LOG_API_USAGE_ONCE("torch.tracer"); in createGraphByTracingWithDict()
140 C10_LOG_API_USAGE_ONCE("torch.tracer"); in createGraphByTracing()
|
| H A D | script_init.cpp | 1613 C10_LOG_API_USAGE_ONCE("torch.script.compile"); in initJitScriptBindings()
1721 C10_LOG_API_USAGE_ONCE("torch.script.class"); in initJitScriptBindings()
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/ |
| H A D | py_rref.cpp | 122 C10_LOG_API_USAGE_ONCE("torch.distributed.rref"); in PyRRef()
184 C10_LOG_API_USAGE_ONCE("torch.distributed.rref.to_here"); in toHere()
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/autograd/ |
| H A D | autograd.cpp | 15 C10_LOG_API_USAGE_ONCE("torch.distributed.autograd.backward"); in backward()
|
| /aosp_15_r20/external/pytorch/binaries/ |
| H A D | core_overhead_benchmark.cc | 28 C10_LOG_API_USAGE_ONCE("bla"); in call()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/xpu/detail/ |
| H A D | XPUHooks.cpp | 13 C10_LOG_API_USAGE_ONCE("aten.init.xpu"); in initXPU()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/mps/ |
| H A D | MPSHooks.mm | 14 C10_LOG_API_USAGE_ONCE("aten.init.mps");
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/ |
| H A D | attention.cu | 693 C10_LOG_API_USAGE_ONCE("torch.sdpa.flash_attention"); in _scaled_dot_product_flash_attention_cuda()
747 C10_LOG_API_USAGE_ONCE("torch.sdpa.flash_attention_cudnn"); in _scaled_dot_product_cudnn_attention_cuda()
799 C10_LOG_API_USAGE_ONCE("torch.sdpa.mem_efficient_attention"); in _scaled_dot_product_efficient_attention_cuda()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | graph_executor.cpp | 581 C10_LOG_API_USAGE_ONCE("torch.graph_executor.run"); in run()
600 C10_LOG_API_USAGE_ONCE("torch.graph_executor.runAsync"); in runAsync()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/detail/ |
| H A D | CUDAHooks.cpp | 100 C10_LOG_API_USAGE_ONCE("aten.init.cuda"); in initCUDA()
|
| /aosp_15_r20/external/pytorch/c10/util/ |
| H A D | Logging.h | 325 #define C10_LOG_API_USAGE_ONCE(...) \ macro
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | NestedTensorImpl.cpp | 177 C10_LOG_API_USAGE_ONCE("torch.NestedTensor"); in NestedTensorImpl()
|
| /aosp_15_r20/external/pytorch/docs/source/notes/ |
| H A D | large_scale_deployments.rst | 99 ``C10_LOG_API_USAGE_ONCE("my_api")`` in C++ or
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/autograd/engine/ |
| H A D | dist_engine.cpp | 85 C10_LOG_API_USAGE_ONCE("torch.autograd.thread_shutdown"); in globalCpuThread()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/ |
| H A D | import.cpp | 255 C10_LOG_API_USAGE_ONCE("torch.jit.load"); in deserialize()
|
| H A D | export_module.cpp | 453 C10_LOG_API_USAGE_ONCE("torch.jit.save"); in serialize()
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | TensorImpl.cpp | 150 C10_LOG_API_USAGE_ONCE("tensor.create"); in TensorImpl()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/ |
| H A D | attention.cpp | 719 C10_LOG_API_USAGE_ONCE("torch.sdpa.math_fallback"); in _scaled_dot_product_attention_math()
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | engine.cpp | 541 C10_LOG_API_USAGE_ONCE("torch.autograd.thread_shutdown"); in thread_main()
|
| /aosp_15_r20/external/pytorch/torch/csrc/ |
| H A D | Module.cpp | 1646 C10_LOG_API_USAGE_ONCE("torch.python.import"); in initModule()
|