Searched refs:DeviceMap (Results 1 – 17 of 17) sorted by relevance
| /external/pytorch/torch/csrc/distributed/autograd/rpc_messages/ |
| D | rpc_with_autograd.h | 22 rpc::DeviceMap deviceMap = {});
32 rpc::DeviceMap deviceMap = {});
58 const rpc::DeviceMap& deviceMap();
93 rpc::DeviceMap deviceMap_;
|
| D | rpc_with_autograd.cpp | 21 rpc::DeviceMap deviceMap) in RpcWithAutograd()
41 rpc::DeviceMap deviceMap) in RpcWithAutograd()
116 rpc::DeviceMap deviceMap; in fromMessage()
173 const rpc::DeviceMap& RpcWithAutograd::deviceMap() { in deviceMap()
|
| /external/pytorch/torch/csrc/distributed/rpc/ |
| D | tensorpipe_agent.h | 86 std::unordered_map<std::string, DeviceMap> device_maps = {},
118 void setDeviceMap(const std::string& workerName, const DeviceMap& deviceMap) { in setDeviceMap()
139 std::unordered_map<std::string, DeviceMap> deviceMaps;
169 std::unordered_map<std::string, DeviceMap> reverseDeviceMaps,
180 const DeviceMap& deviceMap = {}) override;
197 const std::unordered_map<std::string, DeviceMap>& reverseDeviceMaps,
208 DeviceMap getDeviceMap(const WorkerInfo& dest) const override;
331 std::unordered_map<std::string, DeviceMap> reverseDeviceMaps_;
|
| D | rpc_agent.h | 16 using DeviceMap = std::unordered_map<c10::Device, c10::Device>; variable
151 const DeviceMap& deviceMap = {}) = 0;
251 virtual DeviceMap getDeviceMap(const WorkerInfo& dst) const;
|
| D | init.cpp | 542 std::unordered_map<std::string, DeviceMap>, in rpc_init()
549 py::arg("device_maps") = std::unordered_map<std::string, DeviceMap>(), in rpc_init()
580 std::unordered_map<std::string, DeviceMap> reverseDeviceMaps, in rpc_init()
633 (DeviceMap(TensorPipeAgent::*)(const WorkerInfo& dst) const) & in rpc_init()
|
| D | tensorpipe_agent.cpp | 40 const DeviceMap& deviceMap, in getDevicesForTensors()
389 std::unordered_map<std::string, DeviceMap> reverseDeviceMaps, in TensorPipeAgent()
782 const DeviceMap& deviceMap) { in send()
1222 const std::unordered_map<std::string, DeviceMap>& reverseDeviceMaps, in updateGroupMembership()
1403 std::unordered_map<std::string, DeviceMap> deviceMaps; in getDevicesForRemote()
1433 DeviceMap TensorPipeAgent::getDeviceMap(const WorkerInfo& dst) const { in getDeviceMap()
|
| D | rpc_agent.cpp | 313 DeviceMap RpcAgent::getDeviceMap(const WorkerInfo& /* unused */) const { in getDeviceMap()
|
| D | utils.cpp | 178 DeviceMap reverseDeviceMap; in deserializeResponse()
|
| D | request_callback_no_python.cpp | 290 DeviceMap reverseDeviceMap; in processForwardAutogradReq()
|
| /external/pytorch/torch/csrc/distributed/autograd/functions/ |
| D | recvrpc_backward.h | 26 rpc::DeviceMap deviceMap);
44 const rpc::DeviceMap deviceMap_;
|
| D | recvrpc_backward.cpp | 18 rpc::DeviceMap deviceMap) in RecvRpcBackward()
|
| /external/pytorch/torch/csrc/distributed/autograd/ |
| D | utils.h | 34 const rpc::DeviceMap& deviceMap);
47 const rpc::DeviceMap& deviceMap = {});
|
| D | utils.cpp | 55 const rpc::DeviceMap& deviceMap) { in addRecvRpcBackward()
106 const rpc::DeviceMap& deviceMap) { in getMessageWithAutograd()
|
| /external/pytorch/torch/csrc/distributed/rpc/testing/ |
| D | faulty_tensorpipe_agent.h | 46 std::unordered_map<std::string, DeviceMap> reverseDeviceMaps,
55 const DeviceMap& deviceMap = {}) override;
|
| D | faulty_tensorpipe_agent.cpp | 20 std::unordered_map<std::string, DeviceMap> reverseDeviceMaps, in FaultyTensorPipeAgent()
68 const DeviceMap& /* unused */) { in send() argument
|
| D | init.cpp | 76 std::unordered_map<std::string, DeviceMap> reverse_device_maps, in faulty_agent_init()
|
| /external/pytorch/test/cpp/rpc/ |
| D | test_e2e_tensorpipe.cpp | 37 std::unordered_map<std::string, DeviceMap>{}, in buildRpcAgent()
|