Searched refs:ConcretePerOpGpuDevice (Results 1 – 3 of 3) sorted by relevance
57 class ConcretePerOpGpuDevice : public PerOpGpuDevice {59 ConcretePerOpGpuDevice();
899 ConcretePerOpGpuDevice::ConcretePerOpGpuDevice() in ConcretePerOpGpuDevice() function in tensorflow::ConcretePerOpGpuDevice902 void ConcretePerOpGpuDevice::Reinitialize(OpKernelContext* context, in Reinitialize()915 void ConcretePerOpGpuDevice::Reinitialize(OpKernelContext* context, in Reinitialize()925 const Eigen::GpuDevice& ConcretePerOpGpuDevice::device() const { in device()1161 ConcretePerOpGpuDevice* concrete_device = in ReinitializeDevice()1162 static_cast<ConcretePerOpGpuDevice*>(device); in ReinitializeDevice()1172 return new ConcretePerOpGpuDevice(); in MakeGpuDevice()
314 return new ConcretePerOpGpuDevice(); in MakeGpuDevice()324 auto concrete_device = static_cast<ConcretePerOpGpuDevice*>(device); in ReinitializeGpuDevice()