Home
last modified time | relevance | path

Searched defs:rt_input (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_tensor_shape.py177 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 Dragged_getitem.py35 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 Dragged_conversion_ops.py44 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 Dragged_row_lengths_op_test.py111 rt_input, argument
135 def testErrors(self, rt_input, exception, message=None, axis=1): argument
H A Dragged_to_tensor_op_test.py320 rt_input, argument
386 rt_input, argument
H A Dragged_array_ops.py258 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 Dragged_expand_dims_op_test.py107 rt_input, argument
H A Dragged_tile_op_test.py184 rt_input, argument
H A Dragged_where_op.py255 def _nrows(rt_input, out_type): argument
H A Dragged_concat_ops.py302 def _increase_ragged_rank_to(rt_input, ragged_rank, row_splits_dtype): argument
H A Dragged_reduce_op_test.py632 def testReduce(self, ragged_reduce_op, rt_input, axis, keepdims, expected): argument
H A Dragged_math_ops.py478 rt_input, argument
H A Dragged_tensor.py2956 def _get_row_partition_type_tensor_pairs(rt_input): argument
H A Ddynamic_ragged_shape.py1623 def broadcast_to(rt_input, shape: DynamicRaggedShape): argument