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()
218 virtual PerOpGpuDevice* MakeGpuDevice() { return nullptr; } in MakeGpuDevice()
226 PerOpGpuDevice* /*device*/, in ReinitializeGpuDevice() argument
|
| D | op_kernel.h | 599 PerOpGpuDevice* eigen_gpu_device = nullptr;
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.h | 94 PerOpGpuDevice* MakeGpuDevice() override;
96 Status ReinitializeGpuDevice(OpKernelContext* context, PerOpGpuDevice* device,
164 void ReinitializeDevice(OpKernelContext* context, PerOpGpuDevice* device,
|
| D | gpu_device.cc | 749 class ConcretePerOpGpuDevice : public PerOpGpuDevice {
931 PerOpGpuDevice* device, int stream_id, in ReinitializeDevice()
942 PerOpGpuDevice* BaseGPUDevice::MakeGpuDevice() { in MakeGpuDevice()
947 PerOpGpuDevice* device, in ReinitializeGpuDevice()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | renamed_device.h | 81 PerOpGpuDevice* MakeGpuDevice() override { in MakeGpuDevice()
85 Status ReinitializeGpuDevice(OpKernelContext* context, PerOpGpuDevice* device, in ReinitializeGpuDevice()
|