| /external/pytorch/torch/csrc/distributed/autograd/rpc_messages/ |
| D | rpc_with_autograd.cpp | 21 rpc::DeviceMap deviceMap) in RpcWithAutograd() argument
26 deviceMap_(std::move(deviceMap)) { in RpcWithAutograd()
41 rpc::DeviceMap deviceMap) in RpcWithAutograd() argument
48 deviceMap_(std::move(deviceMap)) { in RpcWithAutograd()
63 c10::Dict<std::string, std::string> deviceMap; in toMessageImpl() local
65 deviceMap.insert(mapEntry.first.str(), mapEntry.second.str()); in toMessageImpl()
73 deviceMap}; in toMessageImpl()
116 rpc::DeviceMap deviceMap; in fromMessage() local
118 deviceMap.insert({mapEntry.key(), mapEntry.value()}); in fromMessage()
139 deviceMap); in fromMessage()
[all …]
|
| D | rpc_with_autograd.h | 22 rpc::DeviceMap deviceMap = {});
32 rpc::DeviceMap deviceMap = {});
58 const rpc::DeviceMap& deviceMap();
|
| /external/OpenCSD/decoder/tests/snapshot_parser_lib/source/ |
| D | device_parser.cpp | 73 deviceMap[deviceID] = DevPtr(new DeviceInfo(deviceID, path)); in addDevice()
112 deviceCount = deviceMap.size(); in getDeviceCount()
118 map<int, DevPtr>::const_iterator it(deviceMap.find(id)); in getDevice()
119 if (it == deviceMap.end()) in getDevice()
137 for (it = deviceMap.begin(); it != deviceMap.end(); ++it) in getDeviceList()
|
| /external/pytorch/torch/csrc/distributed/autograd/ |
| D | utils.cpp | 55 const rpc::DeviceMap& deviceMap) { in addRecvRpcBackward() argument
64 autogradMetadata, autogradContext, fromWorkerId, deviceMap); in addRecvRpcBackward()
106 const rpc::DeviceMap& deviceMap) { in getMessageWithAutograd() argument
130 deviceMap); in getMessageWithAutograd()
|
| D | utils.h | 34 const rpc::DeviceMap& deviceMap);
47 const rpc::DeviceMap& deviceMap = {});
|
| /external/pytorch/torch/csrc/distributed/autograd/functions/ |
| D | recvrpc_backward.cpp | 18 rpc::DeviceMap deviceMap) in RecvRpcBackward() argument
22 deviceMap_(std::move(deviceMap)) {} in RecvRpcBackward()
|
| D | recvrpc_backward.h | 26 rpc::DeviceMap deviceMap);
|
| /external/pytorch/torch/csrc/distributed/rpc/ |
| D | tensorpipe_agent.h | 118 void setDeviceMap(const std::string& workerName, const DeviceMap& deviceMap) { in setDeviceMap()
121 deviceMaps[workerName] = deviceMap; in setDeviceMap()
123 for (auto& entry : deviceMap) { in setDeviceMap()
180 const DeviceMap& deviceMap = {}) override;
|
| D | tensorpipe_agent.cpp | 40 const DeviceMap& deviceMap, in getDevicesForTensors() argument
55 const auto deviceIter = deviceMap.find(c10::kCPU); in getDevicesForTensors()
56 if (deviceIter == deviceMap.end()) { in getDevicesForTensors()
63 const auto deviceIter = deviceMap.find(t.device()); in getDevicesForTensors()
65 deviceIter != deviceMap.end(), in getDevicesForTensors()
782 const DeviceMap& deviceMap) { in send() argument
839 if (deviceMap.empty()) { in send()
846 deviceMap, in send()
|
| D | rpc_agent.h | 151 const DeviceMap& deviceMap = {}) = 0;
|
| D | utils.cpp | 179 for (const auto& mapEntry : rpcWithAutograd.deviceMap()) { in deserializeResponse()
|
| D | request_callback_no_python.cpp | 291 for (const auto& mapEntry : rpcWithAutograd.deviceMap()) { in processForwardAutogradReq()
|
| /external/OpenCSD/decoder/tests/snapshot_parser_lib/include/ |
| D | device_parser.h | 98 std::map<int, DevPtr> deviceMap; variable
|
| /external/pytorch/torch/csrc/distributed/rpc/testing/ |
| D | faulty_tensorpipe_agent.h | 55 const DeviceMap& deviceMap = {}) override;
|
| /external/deqp/external/vulkancts/scripts/ |
| D | gen_framework.py | 2547 deviceMap = {}
2566 list = deviceMap.get(currVersion)
2567 deviceMap[currVersion] = list + [ext] if list else [ext]
2596 list = deviceMap.get(currVersion)
2597 deviceMap[currVersion] = list + [ext] if list else [ext]
2601 …(uint32_t coreVersion, ::std::vector<const char*>&%s)" % (" dst" if len(deviceMap) != 0 or isSC el…
2602 "{"] + writeExtensionsForVersions(deviceMap) + [
|