Searched refs:ConvolutionVisitor (Results 1 – 2 of 2) sorted by relevance
64 class ConvolutionVisitor { class188 ~ConvolutionVisitor() = default;190 explicit ConvolutionVisitor(int64 limit_on_batch_size,253 ConvolutionVisitor::ConvolutionVisitor(int64 limit_on_batch_size, in ConvolutionVisitor() function in xla::__anon6418faad0111::ConvolutionVisitor276 bool ConvolutionVisitor::IsConvSuitableForSpaceToBatch( in IsConvSuitableForSpaceToBatch()343 StatusOr<HloInstruction*> ConvolutionVisitor::HaloDuplicateWithSlice( in HaloDuplicateWithSlice()477 StatusOr<ConvolutionVisitor::SpaceNextToBatchDetails>478 ConvolutionVisitor::BringSpaceNextToBatch( in BringSpaceNextToBatch()569 ConvolutionVisitor::IncreaseSpatialSizeOnSpaceToBatchedShape( in IncreaseSpatialSizeOnSpaceToBatchedShape()621 StatusOr<bool> ConvolutionVisitor::Run() { in Run()[all …]
46 class ConvolutionVisitor : public DfsHloVisitorWithDefault { class65 ~ConvolutionVisitor() override = default;68 explicit ConvolutionVisitor( in ConvolutionVisitor() function in xla::__anon537e3dca0111::ConvolutionVisitor93 bool ConvolutionVisitor::Run( in Run()97 ConvolutionVisitor visitor(computation, is_cost_viable, in Run()200 Status ConvolutionVisitor::HandleBatchGroupCount(HloInstruction* convolution) { in HandleBatchGroupCount()416 Status ConvolutionVisitor::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()675 if (ConvolutionVisitor::Run(comp, is_cost_viable_, in Run()