Srinivas Vasudevan 4f341bb742 Add Bessel functions to the public api: ...

- tf.math.special.bessel_i0
    - tf.math.special.bessel_i0e
    - tf.math.special.bessel_i1
    - tf.math.special.bessel_i1e
    - tf.math.special.bessel_k0
    - tf.math.special.bessel_k0e
    - tf.math.special.bessel_k1
    - tf.math.special.bessel_k1e
    - tf.math.special.bessel_j0
    - tf.math.special.bessel_j1
    - tf.math.special.bessel_y0
    - tf.math.special.bessel_y1

PiperOrigin-RevId: 317025879
Change-Id: I5c4407eda6bef0d1659b7a566979c7dbbad4ad83

2020-06-17 21:06:09 -07:00

..

optimization

Add Bessel functions to the public api:

2020-06-17 21:06:09 -07:00

serialization

Add implementation for snapshot dataset v2.

2020-06-01 20:38:27 -07:00

assert_cardinality_test.py

[tf.data] Introducing API for statically asserting the cardinality of a dataset. This API improves user experience when using tf.data with Keras in situations where the cardinality of a file-based dataset is known and can thus be asserted. Having a dataset with a statically known cardinality allows Keras to display accurate progress bar on the first epoch.

2020-02-14 18:28:04 -08:00

assert_next_test.py

[tf.data] Migrating remaining experimental API tests to use TF combinations and performing various minor test cleanup.

2020-01-07 11:17:04 -08:00

auto_shard_dataset_test.py

[tf.data] Modifying auto-sharding to remove unsupported assert_cardinality transformations (instead of triggering an error).

2020-04-25 10:36:21 -07:00

bucket_by_sequence_length_test.py

Rolling back a change that broke backwards compatibility.

2020-03-25 09:57:35 -07:00

BUILD

[tf.data] Implementation of tf.data.experimental.save and tf.data.experimental.load. The former makes it possible to save the output of a dataset, while the latter makes it possible to load a previously saved data.

2020-06-11 15:37:49 -07:00

compression_ops_test.py

Add dataset element compression ops.

2020-05-20 20:08:06 -07:00

copy_to_device_test.py

[tf.data] Adding eager mode tests for prefetch_to_device and copy_to_device APIs.

2020-04-07 18:07:04 -07:00

counter_test.py

…

csv_dataset_test.py

[tf.data] Bug fix: make_csv_dataset should not modify mutable parameters passed into it

2020-05-07 16:54:35 -07:00

dense_to_ragged_batch_test.py

tf.data.experimental.dense_to_ragged_batch to output variable ragged rank.

2020-03-30 03:20:46 -07:00

dense_to_sparse_batch_test.py

…

directed_interleave_dataset_test.py

…

get_single_element_test.py

…

group_by_reducer_test.py

…

group_by_window_test.py

Explicitly call RecordStop when calling group iterator for GroupByWindow dataset, since the current_group_iterator isn't correctly wired up to the output node. See b/154341936 for context.

2020-05-28 20:20:26 -07:00

ignore_errors_test.py

…

io_test.py

[tf.data] Implementation of tf.data.experimental.save and tf.data.experimental.load. The former makes it possible to save the output of a dataset, while the latter makes it possible to load a previously saved data.

2020-06-11 15:37:49 -07:00

make_batched_features_dataset_test.py

…

make_csv_dataset_test.py

…

make_tf_record_dataset_test.py

…

map_and_batch_test.py

…

map_defun_op_test.py

[tf.data] Test cleanup.

2020-01-28 10:02:18 -08:00

matching_files_test.py

Small adjustments on import spacing.

2019-12-19 10:59:32 -08:00

model_dataset_test.py

[tf.data] Fixing tracking of processing time for parallel map when inter-op parallelism is disabled.

2020-06-02 15:31:37 -07:00

non_serializable_test.py

[tf.data] Migrating remaining experimental API tests to use TF combinations and performing various minor test cleanup.

2020-01-07 11:17:04 -08:00

optimize_dataset_test.py

[tf.data] Migrating remaining experimental API tests to use TF combinations and performing various minor test cleanup.

2020-01-07 11:17:04 -08:00

override_threadpool_test.py

Replace list comprehension with generator expressions.

2019-12-16 12:26:12 -08:00

parallel_interleave_test.py

Support None option for experimental_interleave sloppiness.

2020-02-14 16:40:58 -08:00

parse_example_dataset_test.py

Enable op-level dataset determinism configuration for ParseExampleDataset

2020-02-13 11:22:03 -08:00

prefetch_to_device_test.py

[tf.data] Explicitly colocate prefetch dataset op with its input as this collocation only happens automatically in graph mode.

2020-05-29 16:12:11 -07:00

prefetch_with_slack_test.py

…

random_dataset_test.py

Avoid baking random seeds into RandomDataset and SamplingDataset.

2020-03-04 13:46:44 -08:00

reader_dataset_ops_test_base.py

…

rebatch_dataset_test.py

[tf.data] Reenable rebatch_dataset_test in TAP and remove some tests from it to reduce its size. These tests exercise logic that was important to check for in the old rewrite-based RebatchDataset. Now that rebatching is done with its own RebatchDataset kernel, they are no longer relevant.

2020-04-01 17:39:07 -07:00

rejection_resample_test.py

[tf.data] Changing the tf.data.experimental.rejection_resampling implementation to avoid relying on the assumption that a dataset copy produces elements in the same order as the original dataset -- which is not guaranteed to be true (e.g. for shuffled datasets).

2020-01-10 14:30:12 -08:00

replicate_cluster_test.py

…

replicate_test.py

[tf.data] Enabling a newly passing test.

2020-04-09 12:04:42 -07:00

scan_test.py

…

shuffle_and_repeat_test.py

…

sleep_test.py

…

snapshot_test.py

Add implementation for snapshot dataset v2.

2020-06-01 20:38:27 -07:00

sql_dataset_test_base.py

…

sql_dataset_test.py

…

stats_dataset_ops_test.py

Update test post forward compatibility horizon update

2020-03-07 13:45:52 -08:00

stats_dataset_test_base.py

Small adjustments on import spacing.

2019-12-19 10:59:32 -08:00

take_while_test.py

…

tf_record_writer_test.py

…

unique_test.py

…

variant_test.py

Merge pull request #40177 from zhuzilin:fix-map-cardinality

2020-06-08 18:27:01 -07:00

wrap_unwrap_test.py

[tf.data] Migrating remaining experimental API tests to use TF combinations and performing various minor test cleanup.

2020-01-07 11:17:04 -08:00