Searched defs:gradients_size (Results 1 – 2 of 2) sorted by relevance
192 XlaOp AvgPoolGrad(XlaOp out_backprop, absl::Span<const int64> gradients_size, in AvgPoolGrad()
504 gtl::FlatMap<int64, int64> gradients_size; in ComputeGradient() local