Searched defs:eigen_device (Results 1 – 5 of 5) sorted by relevance
34 void TestRequantizeMany(Eigen::ThreadPoolDevice* eigen_device, float input_min, in TestRequantizeMany()114 Eigen::ThreadPoolDevice* eigen_device) { in TestRequantizeManyInNewRange32To8Bit()215 Eigen::ThreadPoolDevice eigen_device(threadpool.AsEigenThreadPool(), in TestRequantizeManyInNewRangeEigenVsNonEigen() local297 Eigen::ThreadPoolDevice eigen_device(threadpool.AsEigenThreadPool(), in TimeRequantizeManyInNewRange() local340 Eigen::ThreadPoolDevice* eigen_device) { in TestFloatToQuantizedInPlaceUsingEigen()384 Eigen::ThreadPoolDevice* eigen_device) { in TestQuantizedToFloatInPlaceUsingEigen()609 Eigen::ThreadPoolDevice eigen_device(threadpool.AsEigenThreadPool(), in TestRequantizeManyInNewRange32To8BitUsingEigen() local640 Eigen::ThreadPoolDevice eigen_device(threadpool.AsEigenThreadPool(), in TestFloatToQuantizedInPlaceUsingEigen() local651 Eigen::ThreadPoolDevice eigen_device(threadpool.AsEigenThreadPool(), in TestOverflowWithEigen() local719 Eigen::ThreadPoolDevice eigen_device(threadpool.AsEigenThreadPool(), in TestQuantizedToFloatInPlaceUsingEigen() local
101 const Device& eigen_device = ctx->eigen_device<Device>(); in Compute() local285 const Device& eigen_device = ctx->eigen_device<Device>(); in Compute() local
31 Eigen::ThreadPoolDevice eigen_device(pool.AsEigenThreadPool(), in TEST() local
1779 const Eigen::ThreadPoolDevice& OpKernelContext::eigen_device() const { in eigen_device() function in tensorflow::OpKernelContext1784 const Eigen::GpuDevice& OpKernelContext::eigen_device() const { in eigen_device() function in tensorflow::OpKernelContext
189 const Eigen::ThreadPoolDevice& TFRTOpKernelContext::eigen_device() const { in eigen_device() function in tensorflow::TFRTOpKernelContext