Searched refs:best_algorithm (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.cc | 302 auto best_algorithm = [&]() -> absl::optional<se::blas::AlgorithmType> { in RunGemm() local
318 if (!best_algorithm) { in RunGemm()
327 *best_algorithm, in RunGemm()
339 static_cast<Eigen::half>(beta), stream, best_algorithm, in RunGemm()
346 static_cast<Eigen::bfloat16>(beta), stream, best_algorithm, in RunGemm()
351 alpha.real(), beta, stream, best_algorithm, in RunGemm()
356 alpha.real(), beta, stream, best_algorithm, in RunGemm()
362 best_algorithm, in RunGemm()
367 static_cast<complex128>(beta), stream, best_algorithm, in RunGemm()
|