Searched refs:GpuConvAlgorithmPicker (Results 1 – 3 of 3) sorted by relevance
336 StatusOr<AutotuneResult> GpuConvAlgorithmPicker::PickBestAlgorithm( in PickBestAlgorithm()428 GpuConvAlgorithmPicker::AutotuneOneConvRunner( in AutotuneOneConvRunner()642 GpuConvAlgorithmPicker::PickBestAlgorithmNoCacheCuda( in PickBestAlgorithmNoCacheCuda()793 GpuConvAlgorithmPicker::PickBestAlgorithmNoCacheRocm( in PickBestAlgorithmNoCacheRocm()906 StatusOr<bool> GpuConvAlgorithmPicker::RunOnInstruction(HloInstruction* instr) { in RunOnInstruction()982 StatusOr<bool> GpuConvAlgorithmPicker::RunOnComputation( in RunOnComputation()999 StatusOr<bool> GpuConvAlgorithmPicker::Run( in Run()
40 class GpuConvAlgorithmPicker : public HloModulePass {45 GpuConvAlgorithmPicker(se::StreamExecutor* stream_exec, in GpuConvAlgorithmPicker() function
813 pipeline.AddPass<GpuConvAlgorithmPicker>(stream_exec, device_allocator); in OptimizeHloPostLayoutAssignment()