Home
last modified time | relevance | path

Searched refs:dev_locality (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_rma_local_test.cc71 DeviceLocality dev_locality; in TEST_F() local
78 attr /*to_alloc_attr*/, &sink_tensor, dev_locality, in TEST_F()
94 attr /*to_alloc_attr*/, &source_tensor, dev_locality, in TEST_F()
114 DeviceLocality dev_locality; in TEST_F() local
121 attr /*to_alloc_attr*/, &sink_tensor, dev_locality, in TEST_F()
139 attr /*to_alloc_attr*/, &source_tensor, dev_locality, in TEST_F()
171 DeviceLocality dev_locality; in TEST_F() local
178 attr /*to_alloc_attr*/, &sink_tensor, dev_locality, in TEST_F()
193 DeviceLocality dev_locality; in TEST_F() local
201 attr /*to_alloc_attr*/, &sink_tensor, dev_locality, in TEST_F()
[all …]
Dthreadpool_device_factory.cc57 DeviceLocality dev_locality; in CreateDevices() local
58 dev_locality.set_numa_node(numa_node); in CreateDevices()
60 options, name, Bytes(256 << 20), dev_locality, in CreateDevices()
Dpermuter_test.cc141 DeviceLocality dev_locality; in Init() local
148 sess_opts, dev_name, mem_limit, dev_locality, cpu_allocator())); in Init()
Dring_gatherer_test.cc149 DeviceLocality dev_locality; in Init() local
156 sess_opts, dev_name, mem_limit, dev_locality, cpu_allocator())); in Init()
Dring_reducer_test.cc172 DeviceLocality dev_locality; in Init() local
179 sess_opts, dev_name, mem_limit, dev_locality, cpu_allocator())); in Init()
Dhierarchical_tree_broadcaster_test.cc234 DeviceLocality dev_locality; in Init() local
241 sess_opts, dev_name, mem_limit, dev_locality, cpu_allocator())); in Init()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc1403 int64 memory_limit, const DeviceLocality& dev_locality, size_t num_tf_gpus, in CreateGPUDevice() argument
1413 int numa_node = dev_locality.numa_node(); in CreateGPUDevice()
1453 options, device_name, static_cast<Bytes>(bytes_limit), dev_locality, in CreateGPUDevice()
1545 DeviceLocality dev_locality; in GetDeviceLocalities() local
1546 dev_locality.set_numa_node(numa_node); in GetDeviceLocalities()
1547 dev_locality.set_bus_id(numa_node + 1); in GetDeviceLocalities()
1550 LocalLinks* links = dev_locality.mutable_links(); in GetDeviceLocalities()
1581 (*localities)[tf_gpu_id] = dev_locality; in GetDeviceLocalities()
1583 << tf_gpu_id << " on bus " << dev_locality.bus_id() in GetDeviceLocalities()
1585 << " DeviceLocality: " << dev_locality.DebugString(); in GetDeviceLocalities()
Dgpu_device.h356 int64 memory_limit, const DeviceLocality& dev_locality,
362 const DeviceLocality& dev_locality, TfGpuId tf_gpu_id,