Searched refs:device_map (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | costmodel_test.cc | 84 std::unordered_map<string, const Graph*> device_map; in TEST() local
85 device_map["DummyDevice1"] = graph1.get(); in TEST()
86 device_map["DummyDevice2"] = graph2.get(); in TEST()
88 collector.BuildCostModel(&cost_model_manager, device_map); in TEST()
|
| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/tpu/ |
| D | tpu_cluster_resolver.py | 127 device_map = collections.defaultdict(list)
134 device_map[host_id].append(core_id)
136 return DeviceDetails(device_map, num_cores)
396 device_details.device_map)
|
| D | tpu_cluster_resolver_test.py | 569 self.assertEqual(device_details.device_map,
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | tf_optimizer_wrapper.cc | 42 std::unordered_map<std::string, tensorflow::DeviceProperties>* device_map) { in DetectDevices() argument
50 tensorflow::DeviceProperties& prop = (*device_map)[device->name()]; in DetectDevices()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | rewrite_utils.cc | 125 std::unordered_map<string, tensorflow::DeviceProperties> device_map; in ApplyRewrites() local
126 tensorflow::grappler::VirtualCluster cluster(device_map); in ApplyRewrites()
|
| /external/chromium-trace/catapult/devil/devil/android/ |
| D | forwarder.py | 243 device_map = instance._device_to_host_port_map
245 for device_serial_and_port, host_port in device_map.items():
248 del device_map[device_serial_and_port]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_variable_runtime_reformatting.cc | 423 auto device_map = devices_attr.getValue(); in HandleReplicateOp() local
425 devices.reserve(device_map.size()); in HandleReplicateOp()
427 for (auto it : device_map) { in HandleReplicateOp()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding.cc | 209 std::map<int64, int64> device_map; in UsedDevices() local
214 device_map[*unique_device] += 1; in UsedDevices()
221 device_map[*unique_device] += 1; in UsedDevices()
227 return device_map; in UsedDevices()
|
| D | hlo_sharding_util.cc | 198 const std::map<int64, int64>& device_map, int64* top_count) { in SelectDominantDevice() argument
201 for (auto& it : device_map) { in SelectDominantDevice()
227 std::map<int64, int64> device_map; in GetMostOccurringDevice() local
232 device_map[it.first] += it.second; in GetMostOccurringDevice()
236 return SelectDominantDevice(device_map, nullptr); in GetMostOccurringDevice()
242 std::map<int64, int64> device_map; in GetDominantDevice() local
249 device_map[it.first] += it.second; in GetDominantDevice()
256 absl::optional<int64> device = SelectDominantDevice(device_map, &count); in GetDominantDevice()
|
| D | hlo_sharding_util.h | 56 const std::map<int64, int64>& device_map, int64* top_count);
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | step_stats_collector.h | 166 const std::unordered_map<string, const Graph*>& device_map);
|
| D | step_stats_collector.cc | 261 const std::unordered_map<string, const Graph*>& device_map) { in BuildCostModel() argument
308 for (const auto& itr : device_map) { in BuildCostModel()
|
| /external/mesa3d/src/broadcom/vulkan/ |
| D | v3dv_device.c | 1502 device_map(struct v3dv_device *device, struct v3dv_device_memory *mem) in device_map() function
1743 VkResult result = device_map(device, mem); in v3dv_MapMemory()
|