Searched refs:AvgPoolDivideByCount (Results 1 – 1 of 1) sorted by relevance
108 XlaOp AvgPoolDivideByCount(XlaOp pooled, absl::Span<const int64> input_size, in AvgPoolDivideByCount() function166 return AvgPoolDivideByCount(pooled, input_size, kernel_size, stride, in AvgPool()233 auto out_backprop_div = AvgPoolDivideByCount( in AvgPoolGrad()