Home
last modified time | relevance | path

Searched refs:UseCustomContractionKernels (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Deigen_contraction_kernel.h54 EIGEN_DEVICE_FUNC EIGEN_DONT_INLINE bool UseCustomContractionKernels();
311 if (!UseCustomContractionKernels()) return;
633 if (UseCustomContractionKernels()) { \
653 if (UseCustomContractionKernels()) { \
674 if (UseCustomContractionKernels()) { \
H A Deigen_contraction_kernel.cc47 EIGEN_DEVICE_FUNC EIGEN_DONT_INLINE bool UseCustomContractionKernels() { in UseCustomContractionKernels() function