Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc114 bool DoGemmWithAlgorithm(MatrixDescriptor lhs_matrix, in DoGemmWithAlgorithm() function
171 CHECK(DoGemmWithAlgorithm<Element>(lhs_matrix, rhs_matrix, output_matrix, in DoGemmAutotune()
216 -> decltype(&DoGemmWithAlgorithm<float>) { in GetGemmWithAlgorithmFn()
219 return &DoGemmWithAlgorithm<Eigen::half>; in GetGemmWithAlgorithmFn()
221 return &DoGemmWithAlgorithm<float>; in GetGemmWithAlgorithmFn()
223 return &DoGemmWithAlgorithm<double>; in GetGemmWithAlgorithmFn()
225 return &DoGemmWithAlgorithm<std::complex<float>>; in GetGemmWithAlgorithmFn()
227 return &DoGemmWithAlgorithm<std::complex<double>>; in GetGemmWithAlgorithmFn()