Searched refs:shape_small (Results 1 – 2 of 2) sorted by relevance
94 auto shape_small = shape_x.size() <= shape_y.size() ? shape_x : shape_y; in GetBroadcastType() local102 if (broadcast_dimensions.size() != shape_small.size()) { in GetBroadcastType()109 shape_large.size() - shape_small.size(), shape_large.size())); in GetBroadcastType()118 auto new_value = shape_small[index_pair.index()]; in GetBroadcastType()
396 auto shape_small = shape_x.size() <= shape_y.size() ? shape_x : shape_y; in GetStaticBroadcastType() local403 if (broadcast_dimensions.size() != shape_small.size()) { in GetStaticBroadcastType()412 auto new_value = shape_small[index_pair.index()]; in GetStaticBroadcastType()