Searched refs:broadcast_rank (Results 1 – 2 of 2) sorted by relevance
463 broadcast_rank = max(shape_x.rank, shape_y.rank)464 shape_x = shape_x.broadcast_to_rank(broadcast_rank)465 shape_y = shape_y.broadcast_to_rank(broadcast_rank)468 for axis in range(broadcast_rank):
966 int64_t broadcast_rank = broadcast_dimensions.size(); in BroadcastInDim() local967 if (operand_shape->rank() != broadcast_rank) { in BroadcastInDim()973 for (int i = 0; i < broadcast_rank; i++) { in BroadcastInDim()987 for (int i = 0; i < broadcast_rank; i++) { in BroadcastInDim()