Searched defs:TensorPipeRpcBackendOptions (Results 1 – 3 of 3) sorted by relevance
79 struct TORCH_API TensorPipeRpcBackendOptions : public RpcBackendOptions { struct88 : RpcBackendOptions(rpc_timeout, std::move(init_method)), in RpcBackendOptions()118 void setDeviceMap(const std::string& workerName, const DeviceMap& deviceMap) { in setDeviceMap()136 int numWorkerThreads;137 const std::optional<std::vector<std::string>> transports;138 const std::optional<std::vector<std::string>> channels;139 std::unordered_map<std::string, DeviceMap> deviceMaps;140 std::vector<c10::Device> devices;
21 : TensorPipeRpcBackendOptions( in TensorPipeRpcBackendOptions() function
46 class TensorPipeRpcBackendOptions(_TensorPipeRpcBackendOptionsBase): class