Searched refs:ReinitializeGpuDevice (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | renamed_device.h | 100 Status ReinitializeGpuDevice(OpKernelContext* context, PerOpGpuDevice* device, in ReinitializeGpuDevice() function
103 return underlying_device_->ReinitializeGpuDevice(context, device, dc, in ReinitializeGpuDevice()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 223 virtual Status ReinitializeGpuDevice(OpKernelContext* /*context*/, in ReinitializeGpuDevice() function
|
| D | op_kernel.cc | 351 Status s = params_->device->ReinitializeGpuDevice( in OpKernelContext()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.h | 107 Status ReinitializeGpuDevice(OpKernelContext* context, PerOpGpuDevice* device,
|
| D | gpu_device.cc | 1175 Status BaseGPUDevice::ReinitializeGpuDevice(OpKernelContext* context, in ReinitializeGpuDevice() function in tensorflow::BaseGPUDevice
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | light_outside_compilation.cc | 320 Status ReinitializeGpuDevice(OpKernelContext* context, PerOpGpuDevice* device, in ReinitializeGpuDevice() function in tensorflow::__anonfdc132ad0311::TfCallbackDevice
|