/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
H A D | ragged_tensor_shape.py | 177 def from_tensor(cls, rt_input, dim_size_dtype=None): argument 475 def broadcast_to(rt_input, shape, broadcast_inner_dimensions=True): argument 505 def _broadcast_to_uniform_shape(rt_input, shape, broadcast_inner_dimensions): argument 515 def _broadcast_to_ragged_shape(rt_input, dst_shape, broadcast_inner_dimensions): argument 602 def _ragged_tile_axis(rt_input, axis, repeats, row_splits_dtype): argument
|
H A D | ragged_getitem.py | 35 def ragged_tensor_getitem(rt_input, key): argument 108 def _ragged_getitem(rt_input, key_list): argument 192 def _slice_ragged_row_dimension(rt_input, row_key): argument 232 def _ragged_getitem_inner_dimensions(rt_input, key_list): argument
|
H A D | ragged_conversion_ops.py | 44 def to_tensor(rt_input, default_value=None, name=None): argument 51 def ragged_to_dense(rt_input, default_value=None, shape=None): argument 136 def to_sparse(rt_input, name=None): argument
|
H A D | ragged_row_lengths_op_test.py | 111 rt_input, argument 135 def testErrors(self, rt_input, exception, message=None, axis=1): argument
|
H A D | ragged_to_tensor_op_test.py | 320 rt_input, argument 386 rt_input, argument
|
H A D | ragged_array_ops.py | 258 def _tile_ragged_values(rt_input, multiples, const_multiples=None): argument 317 def _tile_ragged_splits(rt_input, multiples, const_multiples=None): argument
|
H A D | ragged_expand_dims_op_test.py | 107 rt_input, argument
|
H A D | ragged_tile_op_test.py | 184 rt_input, argument
|
H A D | ragged_where_op.py | 255 def _nrows(rt_input, out_type): argument
|
H A D | ragged_concat_ops.py | 302 def _increase_ragged_rank_to(rt_input, ragged_rank, row_splits_dtype): argument
|
H A D | ragged_reduce_op_test.py | 632 def testReduce(self, ragged_reduce_op, rt_input, axis, keepdims, expected): argument
|
H A D | ragged_math_ops.py | 478 rt_input, argument
|
H A D | ragged_tensor.py | 2956 def _get_row_partition_type_tensor_pairs(rt_input): argument
|
H A D | dynamic_ragged_shape.py | 1623 def broadcast_to(rt_input, shape: DynamicRaggedShape): argument
|