Searched refs:parallel_interleave (Results 1 – 12 of 12) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
H A D | parallel_interleave_test.py | 80 interleave_ops.parallel_interleave( 552 interleave_ops.parallel_interleave( 574 interleave_ops.parallel_interleave(_interleave_fn, cycle_length=1)) 637 interleave_ops.parallel_interleave( 681 interleave_ops.parallel_interleave( 712 interleave_ops.parallel_interleave( 756 interleave_ops.parallel_interleave( 779 interleave_ops.parallel_interleave( 822 interleave_ops.parallel_interleave(_interleave_fn, 1))
|
H A D | auto_shard_dataset_test.py | 145 interleave_ops.parallel_interleave(core_readers.TFRecordDataset, 10)) 149 interleave_ops.parallel_interleave(core_readers.TFRecordDataset, 10)) 170 interleave_ops.parallel_interleave(core_readers.TFRecordDataset, 10)) 175 interleave_ops.parallel_interleave(core_readers.TFRecordDataset, 10)) 196 interleave_ops.parallel_interleave(core_readers.TFRecordDataset, 10)) 230 interleave_ops.parallel_interleave(core_readers.TFRecordDataset, 10)) 308 interleave_ops.parallel_interleave(core_readers.TFRecordDataset, 10)) 427 interleave_ops.parallel_interleave(core_readers.TFRecordDataset, 10)) 707 interleave_ops.parallel_interleave(core_readers.TFRecordDataset, 10))
|
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/ |
H A D | api_def_ParallelInterleaveDataset.pbtxt | 93 The Python API `tf.data.experimental.parallel_interleave` creates instances of 94 this op. `tf.data.experimental.parallel_interleave` is a deprecated API.
|
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
H A D | make_deterministic_test.py | 80 interleave_ops.parallel_interleave(interleave_fn, cycle_length=5)) 245 interleave_ops.parallel_interleave(interleave_fn, cycle_length=5))
|
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/ |
H A D | interleave_ops.py | 29 def parallel_interleave(map_func, function
|
/aosp_15_r20/external/tensorflow/tensorflow/python/data/benchmarks/ |
H A D | interleave_benchmark.py | 68 interleave_ops.parallel_interleave(
|
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
H A D | datasets.py | 152 interleave_ops.parallel_interleave(
|
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ |
H A D | __init__.py | 117 from tensorflow.python.data.experimental.ops.interleave_ops import parallel_interleave
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
H A D | tensorflow.data.experimental.pbtxt | 180 name: "parallel_interleave"
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
H A D | tensorflow.data.experimental.pbtxt | 208 name: "parallel_interleave"
|
/aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/ |
H A D | checkpoint_test.py | 537 interleave_ops.parallel_interleave(self._statefulDatasetFunc, 2))
|
/aosp_15_r20/external/tensorflow/ |
H A D | RELEASE.md | 6158 * Fixes concurrency issue with `tf.data.experimental.parallel_interleave` 8470 * Augment `parallel_interleave` to support 2 kinds of prefetching.
|