Searched refs:IsConvSuitableForSpaceToBatch (Results 1 – 1 of 1) sorted by relevance
180 bool IsConvSuitableForSpaceToBatch(HloInstruction* convolution);343 if (!IsConvSuitableForSpaceToBatch(convolution)) { in ConvolutionVisitor()394 bool ConvolutionVisitor::IsConvSuitableForSpaceToBatch( in IsConvSuitableForSpaceToBatch() function in xla::__anon2cd323230111::ConvolutionVisitor907 !IsConvSuitableForSpaceToBatch(instr)) { in Run()1118 if (IsConvSuitableForSpaceToBatch(consumer)) { in CanPropagate()1889 if (IsConvSuitableForSpaceToBatch(consumer)) { in Propagate()