Searched refs:dst_shape (Results 1 – 1 of 1) sorted by relevance
480 def _broadcast_to_ragged_shape(rt_input, dst_shape, broadcast_inner_dimensions): argument483 if rt_input.shape.ndims is None or dst_shape.rank is None:485 if rt_input.shape.ndims > dst_shape.rank:488 rt_input.ragged_rank >= dst_shape.num_partitioned_dimensions):492 src_shape = src_shape.broadcast_to_rank(dst_shape.rank)495 if dst_shape.rank > rt_input.shape.ndims:496 if rt_input.shape.ndims < dst_shape.num_inner_dimensions + 1:498 rt_input, array_ops.concat([[-1], dst_shape.inner_dim_sizes], axis=0))499 for _ in range(dst_shape.rank - rt_input.shape.ndims):509 rt_input.flat_values.shape.ndims - 1 - dst_shape.num_inner_dimensions)[all …]