Searched refs:ConvolutionVisitor (Results 1 – 1 of 1) sorted by relevance
44 class ConvolutionVisitor : public DfsHloVisitorWithDefault { class64 ~ConvolutionVisitor() override = default;67 explicit ConvolutionVisitor( in ConvolutionVisitor() function in xla::__anon537e3dca0111::ConvolutionVisitor93 bool ConvolutionVisitor::Run( in Run()97 ConvolutionVisitor visitor(computation, is_cost_viable, in Run()201 Status ConvolutionVisitor::HandleBatchGroupCount(HloInstruction* convolution) { in HandleBatchGroupCount()319 Status ConvolutionVisitor::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()546 if (ConvolutionVisitor::Run(comp, is_cost_viable_, in Run()