Searched defs:eigen_device (Results 1 – 4 of 4) 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
91 const Device& eigen_device = ctx->eigen_device<Device>(); in Compute() local270 const Device& eigen_device = ctx->eigen_device<Device>(); in Compute() local
31 Eigen::ThreadPoolDevice eigen_device(pool.AsEigenThreadPool(), in TEST() local
1711 const Eigen::ThreadPoolDevice& OpKernelContext::eigen_device() const { in eigen_device() function in tensorflow::OpKernelContext1716 const Eigen::GpuDevice& OpKernelContext::eigen_device() const { in eigen_device() function in tensorflow::OpKernelContext