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()216 Eigen::ThreadPoolDevice eigen_device(&wrapper, 2 /* num_threads */); in TestRequantizeManyInNewRangeEigenVsNonEigen() local298 Eigen::ThreadPoolDevice eigen_device(&wrapper, 4 /* num_threads */); in TimeRequantizeManyInNewRange() local340 Eigen::ThreadPoolDevice* eigen_device) { in TestFloatToQuantizedInPlaceUsingEigen()384 Eigen::ThreadPoolDevice* eigen_device) { in TestQuantizedToFloatInPlaceUsingEigen()610 Eigen::ThreadPoolDevice eigen_device(&wrapper, 2 /* num_threads */); in TestRequantizeManyInNewRange32To8BitUsingEigen() local641 Eigen::ThreadPoolDevice eigen_device(&wrapper, 2 /* num_threads */); in TestFloatToQuantizedInPlaceUsingEigen() local652 Eigen::ThreadPoolDevice eigen_device(&wrapper, 2 /* num_threads */); in TestOverflowWithEigen() local720 Eigen::ThreadPoolDevice eigen_device(&wrapper, 2 /* num_threads */); in TestQuantizedToFloatInPlaceUsingEigen() local
103 const Device& eigen_device = ctx->eigen_device<Device>(); in Compute() local209 const Device& eigen_device = ctx->eigen_device<Device>(); in Compute() local
33 Eigen::ThreadPoolDevice eigen_device(&wrapper, pool.NumThreads()); in TEST() local
1373 const Eigen::ThreadPoolDevice& OpKernelContext::eigen_device() const { in eigen_device() function in tensorflow::OpKernelContext1378 const Eigen::GpuDevice& OpKernelContext::eigen_device() const { in eigen_device() function in tensorflow::OpKernelContext1384 const Eigen::SyclDevice& OpKernelContext::eigen_device() const { in eigen_device() function in tensorflow::OpKernelContext