Searched refs:PerOpGpuDevice (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 59 class PerOpGpuDevice {
61 virtual ~PerOpGpuDevice() {} in ~PerOpGpuDevice()
212 virtual PerOpGpuDevice* MakeGpuDevice() { return nullptr; } in MakeGpuDevice()
220 PerOpGpuDevice* /*device*/, in ReinitializeGpuDevice() argument
|
| D | op_kernel.h | 617 PerOpGpuDevice* eigen_gpu_device = nullptr;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | renamed_device.h | 95 PerOpGpuDevice* MakeGpuDevice() override { in MakeGpuDevice()
99 Status ReinitializeGpuDevice(OpKernelContext* context, PerOpGpuDevice* device, in ReinitializeGpuDevice()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.h | 81 PerOpGpuDevice* MakeGpuDevice() override;
83 Status ReinitializeGpuDevice(OpKernelContext* context, PerOpGpuDevice* device,
160 void ReinitializeDevice(OpKernelContext* context, PerOpGpuDevice* device,
|
| D | gpu_device.cc | 761 class ConcretePerOpGpuDevice : public PerOpGpuDevice {
1032 PerOpGpuDevice* device, int stream_id, in ReinitializeDevice()
1044 PerOpGpuDevice* BaseGPUDevice::MakeGpuDevice() { in MakeGpuDevice()
1049 PerOpGpuDevice* device, in ReinitializeGpuDevice()
|