Searched refs:common_dims (Results 1 – 1 of 1) sorted by relevance
591 const int common_dims = std::min(shape1.size(), shape2.size()); in MaterializeBroadcastGradientArgs() local592 for (int i = 0; i < common_dims; ++i) { in MaterializeBroadcastGradientArgs()606 for (int i = common_dims; i < shape1.size(); ++i) { in MaterializeBroadcastGradientArgs()611 for (int i = common_dims; i < shape2.size(); ++i) { in MaterializeBroadcastGradientArgs()