Searched refs:GpuConvAlgorithmPicker (Results 1 – 3 of 3) sorted by relevance
35 class GpuConvAlgorithmPicker : public HloModulePass {40 GpuConvAlgorithmPicker(se::StreamExecutor* stream_exec, in GpuConvAlgorithmPicker() function
282 StatusOr<AutotuneResult> GpuConvAlgorithmPicker::PickBestAlgorithm( in PickBestAlgorithm()379 GpuConvAlgorithmPicker::PickBestAlgorithmNoCacheCuda( in PickBestAlgorithmNoCacheCuda()654 GpuConvAlgorithmPicker::PickBestAlgorithmNoCacheRocm( in PickBestAlgorithmNoCacheRocm()767 StatusOr<bool> GpuConvAlgorithmPicker::RunOnInstruction(HloInstruction* instr) { in RunOnInstruction()819 StatusOr<bool> GpuConvAlgorithmPicker::RunOnComputation( in RunOnComputation()836 StatusOr<bool> GpuConvAlgorithmPicker::Run(HloModule* module) { in Run()
478 pipeline.AddPass<GpuConvAlgorithmPicker>(stream_exec, device_allocator); in OptimizeHloPostLayoutAssignment()