Searched refs:num_parallel_batches (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | batching.py | 155 num_parallel_batches=None, argument
191 if num_parallel_batches is None and num_parallel_calls is None:
193 elif num_parallel_batches is not None and num_parallel_calls is None:
194 num_parallel_calls = batch_size * num_parallel_batches
195 elif num_parallel_batches is not None and num_parallel_calls is not None:
215 num_parallel_batches=None, argument
252 if num_parallel_batches is None and num_parallel_calls is None:
254 elif num_parallel_batches is not None and num_parallel_calls is None:
255 num_parallel_calls = batch_size * num_parallel_batches
256 elif num_parallel_batches is not None and num_parallel_calls is not None:
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | map_and_batch_dataset_serialization_test.py | 46 num_parallel_batches = 2
58 num_parallel_batches=num_parallel_batches,
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | map_and_batch_test.py | 49 num_parallel_calls=[None, 1, 2], num_parallel_batches=None) +
51 num_parallel_calls=None, num_parallel_batches=10)))
52 def testMapAndBatch(self, num_parallel_calls, num_parallel_batches): argument
70 num_parallel_batches=num_parallel_batches))
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ExperimentalMapAndBatchDataset.pbtxt | 52 to `batch_size * num_parallel_batches` copies of `f` in parallel.
|
| D | api_def_MapAndBatchDataset.pbtxt | 52 to `batch_size * num_parallel_batches` copies of `f` in parallel.
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.experimental.pbtxt | 193 …argspec: "args=[\'map_func\', \'batch_size\', \'num_parallel_batches\', \'drop_remainder\', \'num_…
197 …argspec: "args=[\'map_func\', \'batch_size\', \'num_parallel_batches\', \'drop_remainder\', \'num_…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.experimental.pbtxt | 169 …argspec: "args=[\'map_func\', \'batch_size\', \'num_parallel_batches\', \'drop_remainder\', \'num_…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.td | 1124 to `batch_size * num_parallel_batches` copies of `f` in parallel.
|