Searched refs:RunGemm (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.h | 79 Status RunGemm(
|
| D | gemm_thunk.cc | 67 return RunGemm(config_, lhs_data, rhs_data, output_data, params.stream, in ExecuteOnStream()
171 Status RunGemm(const GpuGemmConfig &gemm_config, in RunGemm() function
|
| D | gemm_algorithm_picker.cc | 134 CHECK(RunGemm(config, lhs_buffer, rhs_buffer, output_buffer, stream, in DoUncachedGemmAutotune()
|