Searched refs:setCurrentKey (Results 1 – 4 of 4) sorted by relevance
16 void setCurrentKey(std::string key);
17 void RemoteProfilerManager::setCurrentKey(std::string key) { in setCurrentKey() function in torch::distributed::rpc::RemoteProfilerManager
37 remoteProfilerManager.setCurrentKey(rpcAsyncJitKey); in rpcTorchscript()
824 inst.setCurrentKey(key); in rpc_init()