Home
last modified time | relevance | path

Searched defs:gemm_config (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_algorithm_picker.cc232 const HloInstruction* instr, const GemmBackendConfig& gemm_config, in DoGemmAutotune()
286 GemmBackendConfig gemm_config = in RunOnInstruction() local
Dgemm_rewriter.cc63 GemmBackendConfig gemm_config; in HandleDot() local
Dgemm_thunk.cc171 Status RunGemm(const GpuGemmConfig &gemm_config, in RunGemm()