Searched refs:EigenGpuStreamDevice (Results 1 – 1 of 1) sorted by relevance
119 class EigenGpuStreamDevice : public ::Eigen::StreamInterface { class121 EigenGpuStreamDevice() in EigenGpuStreamDevice() function in tensorflow::EigenGpuStreamDevice126 ~EigenGpuStreamDevice() override {} in ~EigenGpuStreamDevice()233 TF_DISALLOW_COPY_AND_ASSIGN(EigenGpuStreamDevice);900 : stream_device_(std::make_unique<EigenGpuStreamDevice>()) {} in ConcretePerOpGpuDevice()910 static_cast<EigenGpuStreamDevice*>(stream_device_.get()) in Reinitialize()920 static_cast<EigenGpuStreamDevice*>(stream_device_.get()) in Reinitialize()926 return static_cast<EigenGpuStreamDevice*>(stream_device_.get())->device(); in device()