Edward Loper 151c540727 Fix issue where tf.ragged.stack([tensor]) would return a Tensor rather than a RaggedTensor. ...

Also updated docs to clarify that the result will be a `Tensor` if the rank of result is 1 (since `RaggedTensor` can only be used when `rank>1`).  E.g., `tf.ragged.stack([1, 2, 3])` returns a 1D `Tensor`, not a `RaggedTensor`.

PiperOrigin-RevId: 357983061
Change-Id: I41b69f395142162357bcd8d7385221b5fb2267c1

2021-02-17 10:49:01 -08:00

..

__init__.py

…

BUILD

PY2 removal cleanup

2021-01-20 17:11:44 -08:00

convert_to_tensor_or_ragged_tensor_op_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_array_ops.py

Fix support of ragged expand_dims on other value types (experimental api).

2020-10-26 10:31:00 -07:00

ragged_batch_gather_op_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_batch_gather_ops.py

…

ragged_batch_gather_with_default_op.py

…

ragged_boolean_mask_op_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_concat_op_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_concat_ops.py

Fix issue where tf.ragged.stack([tensor]) would return a Tensor rather than a RaggedTensor.

2021-02-17 10:49:01 -08:00

ragged_config.py

…

ragged_const_op_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_constant_value_op_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_conversion_ops.py

Added gradients for RaggedTensorToVariant and RaggedTensorFromVariant. (This allows gradients to pass through map_fn when it is applied to ragged tensors.)

2020-10-14 10:09:39 -07:00

ragged_cross_op_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_dispatch_test.py

Commeting out subtests that are failing due to JIRA ticket 236756, and also removing the no_rocm tag from the tests that contain those subtests

2020-09-24 19:40:56 +00:00

ragged_dispatch.py

Fix error message when registering a ragged op dispatcher to an op that's not an exported symbol.

2020-10-12 14:49:22 -07:00

ragged_dynamic_partition_op_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_eager_test.py

…

ragged_expand_dims_op_test.py

…

ragged_factory_ops.py

Do not add validation ops in tf.ragged.placeholder.

2020-09-14 18:09:58 -07:00

ragged_from_sparse_op_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_from_tensor_op_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_functional_ops.py

Improvements to the documentation for tf.ragged.map_flat_values.

2020-11-03 06:38:00 -08:00

ragged_gather_nd_op_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_gather_op_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_gather_ops.py

…

ragged_getitem_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_getitem.py

fix typos in python directory

2020-10-29 16:21:24 +03:00

ragged_map_flat_values_op_test.py

Improve error message when tf.ragged.map_flat_values is called with a function that doesn't preserve the outer dimension size of ragged inputs.

2020-08-10 14:34:40 -07:00

ragged_map_fn_op_test.py

Fix segmentation fault in tf.map_fn when fn_output_spec is a RaggedTensorSpec and the input tensor has shape [0, ...].

2020-08-25 08:18:15 -07:00

ragged_map_ops.py

…

ragged_math_ops.py

Implement support for keepdims=True on ragged reduce operations.

2020-10-07 09:28:29 -07:00

ragged_merge_dims_op_test.py

Update RaggedTensor.merge_dims to return the ragged tensor as-is if outer_axis==inner_axis. I.e.: rt.merge_dims(x, x) == rt. (Previously, it raised an exception for outer_axis==inner_axis.)

2020-11-09 10:36:27 -08:00

ragged_one_hot_op_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_operators_test.py

Update RaggedTensor.__eq__ and RaggedTensor.__ne__ to compare by value (with broadcasting), rather than by identity. (This matches a change that was made for tf.Tensor going from TF1 to TF2, where Tensor.__eq__ compares by value rather than identity.

2020-09-25 09:08:54 -07:00

ragged_operators.py

Update RaggedTensor.__eq__ and RaggedTensor.__ne__ to compare by value (with broadcasting), rather than by identity. (This matches a change that was made for tf.Tensor going from TF1 to TF2, where Tensor.__eq__ compares by value rather than identity.

2020-09-25 09:08:54 -07:00

ragged_ops.py

…

ragged_placeholder_op_test.py

Do not add validation ops in tf.ragged.placeholder.

2020-09-14 18:09:58 -07:00

ragged_print_op_test.py

…

ragged_range_op_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_rank_op_test.py

…

ragged_reduce_op_test.py

Implement support for keepdims=True on ragged reduce operations.

2020-10-07 09:28:29 -07:00

ragged_reverse_op_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_row_lengths_op_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_row_splits_to_segment_ids_op_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_segment_ids_to_row_splits_op_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_segment_op_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_size_op_test.py

…

ragged_squeeze_op_test.py

…

ragged_squeeze_op.py

…

ragged_stack_op_test.py

Fix issue where tf.ragged.stack([tensor]) would return a Tensor rather than a RaggedTensor.

2021-02-17 10:49:01 -08:00

ragged_string_ops.py

Internal symbol name change.

2021-01-05 14:15:45 -08:00

ragged_tensor_bounding_shape_op_test.py

…

ragged_tensor_shape_test.py

In RaggedTensorDynamicShape.from_tensor: Check for uniform_row_lengths.

2020-12-17 20:05:38 -08:00

ragged_tensor_shape.py

In RaggedTensorDynamicShape.from_tensor: Check for uniform_row_lengths.

2020-12-17 20:05:38 -08:00

ragged_tensor_supported_values_test.py

Fix support of ragged expand_dims on other value types (experimental api).

2020-10-26 10:31:00 -07:00

ragged_tensor_test_ops.py

Experimental support of _add_supported_value_type for RaggedTensor.

2020-09-02 10:06:20 -07:00

ragged_tensor_test.py

Added gradients for RaggedTensorToVariant and RaggedTensorFromVariant. (This allows gradients to pass through map_fn when it is applied to ragged tensors.)

2020-10-14 10:09:39 -07:00

ragged_tensor_value.py

…

ragged_tensor.py

Add TypeSpec to JSON encoder/Decoder

2020-12-02 16:26:53 -08:00

ragged_tile_op_test.py

…

ragged_to_sparse_op_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_to_tensor_op_test.py

In RaggedTensor.to_tensor(), allow shape to be a list whose elements are ints or tensors.

2020-10-21 07:56:58 -07:00

ragged_util_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

ragged_util.py

…

ragged_where_op_test.py

Update tf.where to handle RaggedTensors in TF2.

2020-09-09 07:25:01 -07:00

ragged_where_op.py

Update tf.where to handle RaggedTensors in TF2.

2020-09-09 07:25:01 -07:00

row_partition_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

row_partition.py

…

segment_id_ops.py

…

string_ngrams_op_test.py

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

strings_reduce_join_op_test.py

…