Searched refs:map_and_batch (Results 1 – 21 of 21) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | map_and_batch_test.py | 71 batching.map_and_batch(
137 batching.map_and_batch(
165 batching.map_and_batch(lambda x: array_ops.reshape(x * x, [1]), 4)))
183 batching.map_and_batch(lambda x: x, batch_size=100))
216 batching.map_and_batch(
256 batching.map_and_batch(_sparse, 5))
282 dataset = dataset.apply(batching.map_and_batch(lambda x: x, 14))
306 dataset = dataset.apply(batching.map_and_batch(lambda x: x, batch_size))
333 1000).apply(batching.map_and_batch(_map_fn, batch_size=100))
366 batching.map_and_batch(
[all …]
|
| D | rebatch_dataset_test.py | 145 batching.map_and_batch(
|
| D | stats_dataset_ops_test.py | 226 batching.map_and_batch(
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | map_and_batch_dataset_serialization_test.py | 49 batching.map_and_batch(
76 batching.map_and_batch(
96 batching.map_and_batch(map_fn, 5))
|
| D | numa_map_and_batch_dataset_serialization_test.py | 48 batching.map_and_batch(
78 batching.map_and_batch(
|
| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| D | batching.py | 232 def map_and_batch(map_func, function
272 return batching.map_and_batch(map_func, batch_size, num_parallel_batches,
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | map_vectorization_test.py | 421 dataset = dataset.apply(batching.map_and_batch(map_fn, 100))
450 batching.map_and_batch(map_fn, 2, 12, drop_remainder=True))
534 unoptimized = base_dataset.apply(batching.map_and_batch(lambda x: x, 2))
553 unoptimized = base_dataset.apply(batching.map_and_batch(map_fn, 2))
|
| D | make_numa_aware_test.py | 34 batching.map_and_batch(lambda x: x * x, 10))
|
| D | optimize_dataset_test.py | 66 batching.map_and_batch(map_fn, 1))
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | quantization_mnist_test.py | 192 data.experimental.map_and_batch(
206 data.experimental.map_and_batch(
|
| /external/tensorflow/tensorflow/contrib/data/ |
| D | __init__.py | 81 from tensorflow.contrib.data.python.ops.batching import map_and_batch
|
| /external/tensorflow/tensorflow/python/data/experimental/ |
| D | __init__.py | 85 from tensorflow.python.data.experimental.ops.batching import map_and_batch
|
| /external/tensorflow/tensorflow/python/data/experimental/benchmarks/ |
| D | map_and_batch_benchmark.py | 55 dataset = dataset.apply(batching.map_and_batch(
|
| D | autotune_benchmark.py | 82 batching.map_and_batch(
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/ |
| D | imagenet_input.py | 191 tf.data.experimental.map_and_batch(
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.experimental.pbtxt | 152 name: "map_and_batch"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.experimental.pbtxt | 152 name: "map_and_batch"
|
| /external/tensorflow/tensorflow/contrib/distribute/ |
| D | README.md | 179 Specifically, we found using [`map_and_batch`](https://www.tensorflow.org/performance/datasets_perf…
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | readers.py | 301 dataset = dataset.apply(batching.map_and_batch(
|
| D | batching.py | 674 def map_and_batch(map_func, function
|
| /external/tensorflow/ |
| D | RELEASE.md | 78 …led by configuring `tf.data.experimental.OptimizationOptions` to set `map_and_batch = False` or `n…
|