Searched refs:DoGemmWithAlgorithm (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | matmul_utils.cc | 423 Status DoGemmWithAlgorithm(int64_t batch_size, int64_t m, int64_t n, int64_t k, in DoGemmWithAlgorithm() function
465 return DoGemmWithAlgorithm<Input, Input>(batch_size, m, n, k, lhs, rhs, in DoGemm()
516 return DoGemmWithAlgorithm<int8_t, int32_t>( in RunGemm()
|