Home
last modified time | relevance | path

Searched refs:RemoteProfilerManager (Results 1 – 9 of 9) sorted by relevance

/external/pytorch/torch/csrc/distributed/rpc/profiler/
Dremote_profiler_manager.cpp11 RemoteProfilerManager::currentThreadLocalKey_ = std::nullopt;
12 /*static */ RemoteProfilerManager& RemoteProfilerManager::getInstance() { in getInstance()
13 static RemoteProfilerManager* handler = new RemoteProfilerManager(); in getInstance()
17 void RemoteProfilerManager::setCurrentKey(std::string key) { in setCurrentKey()
20 if (RemoteProfilerManager::currentThreadLocalKey_) { in setCurrentKey()
28 bool RemoteProfilerManager::isCurrentKeySet() const { in isCurrentKeySet()
32 void RemoteProfilerManager::unsetCurrentKey() { in unsetCurrentKey()
36 void RemoteProfilerManager::eraseKey(const ProfilingId& globallyUniqueId) { in eraseKey()
43 std::string RemoteProfilerManager::retrieveRPCProfilingKey( in retrieveRPCProfilingKey()
51 ProfilingId RemoteProfilerManager::getNextProfilerId() { in getNextProfilerId()
[all …]
Dremote_profiler_manager.h11 class TORCH_API RemoteProfilerManager {
14 static RemoteProfilerManager& getInstance();
40 RemoteProfilerManager(const RemoteProfilerManager& other) = delete;
41 RemoteProfilerManager operator=(const RemoteProfilerManager& other) = delete;
42 RemoteProfilerManager(RemoteProfilerManager&&) = delete;
43 RemoteProfilerManager& operator=(RemoteProfilerManager&&) = delete;
46 RemoteProfilerManager();
47 ~RemoteProfilerManager() = default;
/external/pytorch/torch/csrc/distributed/rpc/
Dtorchscript_functions.cpp24 !torch::distributed::rpc::RemoteProfilerManager::getInstance() in rpcTorchscript()
36 torch::distributed::rpc::RemoteProfilerManager::getInstance(); in rpcTorchscript()
Dinit.cpp819 RemoteProfilerManager, in rpc_init()
820 std::unique_ptr<RemoteProfilerManager, py::nodelete>>( in rpc_init()
823 auto& inst = RemoteProfilerManager::getInstance(); in rpc_init()
Dutils.cpp43 auto& remoteProfilerManager = RemoteProfilerManager::getInstance(); in processRemoteProfiledEvents()
/external/pytorch/torch/distributed/rpc/
D__init__.py66 RemoteProfilerManager,
Dapi.py29 RemoteProfilerManager,
961 RemoteProfilerManager.set_current_profiling_key(rpc_profiling_key)
/external/pytorch/torch/csrc/distributed/autograd/
Dutils.cpp84 torch::distributed::rpc::RemoteProfilerManager::getInstance(); in getMessageWithProfiling()
/external/pytorch/torch/_C/
D_distributed_rpc.pyi180 class RemoteProfilerManager: