Searched refs:index_pair (Results 1 – 3 of 3) sorted by relevance
116 for (auto index_pair : llvm::enumerate(broadcast_dimensions)) { in GetBroadcastType() local117 auto old_value = out_shape[index_pair.value()]; in GetBroadcastType()118 auto new_value = shape_small[index_pair.index()]; in GetBroadcastType()120 out_shape[index_pair.value()] = new_value; in GetBroadcastType()
320 for (auto index_pair : llvm::enumerate(paddings_value.getIntValues())) { in FoldOperandsPermutation() local321 size_t outer_idx = index_pair.index() / 2; in FoldOperandsPermutation()322 size_t inner_idx = index_pair.index() % 2; in FoldOperandsPermutation()325 index_pair.value().getSExtValue(); in FoldOperandsPermutation()
410 for (auto index_pair : llvm::enumerate(broadcast_dimensions)) { in GetStaticBroadcastType() local411 auto old_value = out_shape[index_pair.value()]; in GetStaticBroadcastType()412 auto new_value = shape_small[index_pair.index()]; in GetStaticBroadcastType()413 out_shape[index_pair.value()] = std::max(old_value, new_value); in GetStaticBroadcastType()