Searched refs:map_and_batch (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | map_and_batch_test.py | 66 batching.map_and_batch(
125 batching.map_and_batch(
145 batching.map_and_batch(lambda x: array_ops.reshape(x * x, [1]), 4)))
156 batching.map_and_batch(lambda x: x, batch_size=100))
182 batching.map_and_batch(
214 batching.map_and_batch(_sparse, 5))
233 dataset = dataset.apply(batching.map_and_batch(lambda x: x, 14))
250 dataset = dataset.apply(batching.map_and_batch(lambda x: x, batch_size))
270 1000).apply(batching.map_and_batch(_map_fn, batch_size=100))
289 batching.map_and_batch(
[all …]
|
| D | stats_dataset_ops_test.py | 386 batching.map_and_batch(
|
| D | optimize_dataset_test.py | 63 batching.map_and_batch(map_fn, 1))
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | map_and_batch_dataset_serialization_test.py | 55 batching.map_and_batch(
81 batching.map_and_batch(
100 batching.map_and_batch(map_fn, 5))
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | map_vectorization_test.py | 505 dataset = dataset.apply(batching.map_and_batch(map_fn, 100))
528 batching.map_and_batch(map_fn, 2, 12, drop_remainder=True))
645 unoptimized = base_dataset.apply(batching.map_and_batch(lambda x: x, 2))
663 unoptimized = base_dataset.apply(batching.map_and_batch(map_fn, 2))
|
| /external/tensorflow/tensorflow/python/data/experimental/benchmarks/ |
| D | map_and_batch_benchmark.py | 51 batching.map_and_batch(lambda _: dense_value, batch_size))
|
| /external/tensorflow/tensorflow/python/data/experimental/ |
| D | __init__.py | 100 from tensorflow.python.data.experimental.ops.batching import map_and_batch
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | batching.py | 213 def map_and_batch(map_func, function
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.experimental.pbtxt | 168 name: "map_and_batch"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.experimental.pbtxt | 192 name: "map_and_batch"
|
| /external/tensorflow/ |
| D | RELEASE.md | 2721 `tf.data.experimental.OptimizationOptions` to set `map_and_batch =
|