Searched refs:GetGpuGemmConfig (Results 1 – 4 of 4) sorted by relevance
44 GpuGemmConfig GetGpuGemmConfig(const HloInstruction* gemm);
31 GpuGemmConfig config = GetGpuGemmConfig(inst); in BuildGemmThunk()
118 GpuGemmConfig config = GetGpuGemmConfig(gemm); in DoUncachedGemmAutotune()
36 GpuGemmConfig GetGpuGemmConfig(const HloInstruction *gemm) { in GetGpuGemmConfig() function