Searched refs:PerOpGpuDevice (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 60 class PerOpGpuDevice {
62 virtual ~PerOpGpuDevice() {} in ~PerOpGpuDevice()
195 virtual PerOpGpuDevice* MakeGpuDevice() { return nullptr; } in MakeGpuDevice()
203 PerOpGpuDevice* /*device*/, in ReinitializeGpuDevice() argument
|
| D | op_kernel.h | 501 PerOpGpuDevice* eigen_gpu_device = nullptr;
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.h | 85 PerOpGpuDevice* MakeGpuDevice() override;
87 void ReinitializeGpuDevice(OpKernelContext* context, PerOpGpuDevice* device,
124 void ReinitializeDevice(OpKernelContext* context, PerOpGpuDevice* device,
|
| D | gpu_device.cc | 646 class ConcretePerOpGpuDevice : public PerOpGpuDevice {
812 PerOpGpuDevice* device, int stream_id, in ReinitializeDevice()
823 PerOpGpuDevice* BaseGPUDevice::MakeGpuDevice() { in MakeGpuDevice()
828 PerOpGpuDevice* device, in ReinitializeGpuDevice()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | renamed_device.h | 76 PerOpGpuDevice* MakeGpuDevice() override { in MakeGpuDevice()
80 void ReinitializeGpuDevice(OpKernelContext* context, PerOpGpuDevice* device, in ReinitializeGpuDevice()
|