Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_rma_local_test.cc64 DeviceLocality dev_locality; in TEST_F() local
71 attr /*to_alloc_attr*/, &sink_tensor, dev_locality, in TEST_F()
87 attr /*to_alloc_attr*/, &source_tensor, dev_locality, in TEST_F()
107 DeviceLocality dev_locality; in TEST_F() local
114 attr /*to_alloc_attr*/, &sink_tensor, dev_locality, in TEST_F()
132 attr /*to_alloc_attr*/, &source_tensor, dev_locality, in TEST_F()
Dthreadpool_device_factory.cc51 DeviceLocality dev_locality; in CreateDevices() local
52 dev_locality.set_numa_node(numa_node); in CreateDevices()
54 options, name, Bytes(256 << 20), dev_locality, in CreateDevices()
Dring_gatherer_test.cc143 DeviceLocality dev_locality; in Init() local
150 sess_opts, dev_name, mem_limit, dev_locality, cpu_allocator())); in Init()
Dring_reducer_test.cc165 DeviceLocality dev_locality; in Init() local
172 sess_opts, dev_name, mem_limit, dev_locality, cpu_allocator())); in Init()
Dhierarchical_tree_broadcaster_test.cc225 DeviceLocality dev_locality; in Init() local
232 sess_opts, dev_name, mem_limit, dev_locality, cpu_allocator())); in Init()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc1184 int64 memory_limit, const DeviceLocality& dev_locality, in CreateGPUDevice() argument
1193 int numa_node = dev_locality.numa_node(); in CreateGPUDevice()
1219 options, device_name, static_cast<Bytes>(bytes_limit), dev_locality, in CreateGPUDevice()
1307 DeviceLocality dev_locality; in GetDeviceLocalities() local
1308 dev_locality.set_numa_node(numa_node); in GetDeviceLocalities()
1309 dev_locality.set_bus_id(numa_node + 1); in GetDeviceLocalities()
1312 LocalLinks* links = dev_locality.mutable_links(); in GetDeviceLocalities()
1343 (*localities)[tf_gpu_id] = dev_locality; in GetDeviceLocalities()
1345 << tf_gpu_id << " on bus " << dev_locality.bus_id() in GetDeviceLocalities()
1347 << " DeviceLocality: " << dev_locality.DebugString(); in GetDeviceLocalities()
Dgpu_device.h302 int64 memory_limit, const DeviceLocality& dev_locality,
307 const DeviceLocality& dev_locality, TfGpuId tf_gpu_id,