Searched refs:broadcast_rank (Results 1 – 2 of 2) sorted by relevance
464 broadcast_rank = max(shape_x.rank, shape_y.rank)465 shape_x = shape_x.broadcast_to_rank(broadcast_rank)466 shape_y = shape_y.broadcast_to_rank(broadcast_rank)469 for axis in range(broadcast_rank):
900 tensorflow::int64 broadcast_rank = broadcast_dimensions.size(); in BroadcastInDim() local901 if (operand_shape->rank() != broadcast_rank) { in BroadcastInDim()907 for (int i = 0; i < broadcast_rank; i++) { in BroadcastInDim()921 for (int i = 0; i < broadcast_rank; i++) { in BroadcastInDim()