Searched refs:common_dims (Results 1 – 2 of 2) sorted by relevance
153 llvm::SmallVector<int64_t, 4> common_dims;170 common_dims.resize(common_rank, kUninitialized);184 int64_t &common_dim = common_dims[i];
645 const int common_dims = std::min(shape1.size(), shape2.size()); in MaterializeBroadcastGradientArgs() local646 for (int i = 0; i < common_dims; ++i) { in MaterializeBroadcastGradientArgs()660 for (int i = common_dims, end = shape1.size(); i < end; ++i) { in MaterializeBroadcastGradientArgs()665 for (int i = common_dims, end = shape2.size(); i < end; ++i) { in MaterializeBroadcastGradientArgs()