Home
last modified time | relevance | path

Searched refs:set_tensorflow_gpu_device_info (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h162 void set_tensorflow_gpu_device_info(GpuDeviceInfo* g) { in set_tensorflow_gpu_device_info() function
/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
Dpluggable_device.cc191 set_tensorflow_gpu_device_info(pluggable_device_info_); in Init()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.cc398 set_tensorflow_gpu_device_info(gpu_device_info.get()); in GetDeviceContextLocked()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_rma_distributed_test.cc346 dst_device->set_tensorflow_gpu_device_info(&gpu_device_info_); in MaybeSetGPUDevice()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc483 set_tensorflow_gpu_device_info(gpu_device_info_); in Init()