Searched refs:broadcast_to_rank (Results 1 – 2 of 2) sorted by relevance
234 def broadcast_to_rank(self, rank): member in RaggedTensorDynamicShape429 shape_x = shape_x.broadcast_to_rank(broadcast_rank)430 shape_y = shape_y.broadcast_to_rank(broadcast_rank)492 src_shape = src_shape.broadcast_to_rank(dst_shape.rank)
115 broadcasted_shape = shape.broadcast_to_rank(rank)