Searched refs:GemmAlgorithmPicker (Results 1 – 3 of 3) sorted by relevance
30 class GemmAlgorithmPicker : public HloModulePass {32 GemmAlgorithmPicker(se::StreamExecutor* stream_exec, in GemmAlgorithmPicker() function
315 StatusOr<bool> GemmAlgorithmPicker::Run(HloModule* module) { in Run()
178 post_pipeline.AddPass<GemmAlgorithmPicker>(stream_exec, device_allocator); in OptimizeHloPostLayoutAssignment()