Searched refs:pooled_type (Results 1 – 1 of 1) sorted by relevance
2197 RankedTensorType pooled_type = in AvgPoolDivideByCount() local2199 Type element_type = pooled_type.getElementType(); in AvgPoolDivideByCount()2215 loc, pooled_type, pooled, divisor, scalar_broadcast_dims); in AvgPoolDivideByCount()2233 loc, pooled_type, in AvgPoolDivideByCount()2244 result = rewriter.create<mhlo::DivOp>(loc, pooled_type, pooled, divisor); in AvgPoolDivideByCount()