Searched refs:GpuGemmConfig (Results 1 – 5 of 5) sorted by relevance
37 struct GpuGemmConfig { struct44 GpuGemmConfig GetGpuGemmConfig(const HloInstruction* gemm); argument51 GemmThunk(ThunkInfo thunk_info, GpuGemmConfig config,63 const GpuGemmConfig config_;80 const GpuGemmConfig& gemm_config, se::DeviceMemoryBase lhs_buffer,
36 GpuGemmConfig GetGpuGemmConfig(const HloInstruction *gemm) { in GetGpuGemmConfig()37 GpuGemmConfig config; in GetGpuGemmConfig()46 GemmThunk::GemmThunk(ThunkInfo thunk_info, GpuGemmConfig config, in GemmThunk()171 Status RunGemm(const GpuGemmConfig &gemm_config, in RunGemm()
31 GpuGemmConfig config = GetGpuGemmConfig(inst); in BuildGemmThunk()
118 GpuGemmConfig config = GetGpuGemmConfig(gemm); in DoUncachedGemmAutotune()
1225 GpuGemmConfig config; in EmitGemmThunkFromMlir()1256 GpuGemmConfig config; in EmitGemmThunkFromMlir()