Searched refs:num_parallel_batches (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | batching.py | 616 num_parallel_batches=None, argument
652 if num_parallel_batches is None and num_parallel_calls is None:
654 elif num_parallel_batches is not None and num_parallel_calls is None:
655 num_parallel_calls = batch_size * num_parallel_batches
656 elif num_parallel_batches is not None and num_parallel_calls is not None:
676 num_parallel_batches=None, argument
716 if num_parallel_batches is None and num_parallel_calls is None:
718 elif num_parallel_batches is not None and num_parallel_calls is None:
719 num_parallel_calls = batch_size * num_parallel_batches
720 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 | 40 num_parallel_batches = 2
52 num_parallel_batches=num_parallel_batches,
|
| D | numa_map_and_batch_dataset_serialization_test.py | 39 num_parallel_batches = 2
51 num_parallel_batches=num_parallel_batches,
|
| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| D | batching.py | 234 num_parallel_batches=None, argument
272 return batching.map_and_batch(map_func, batch_size, num_parallel_batches,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.experimental.pbtxt | 153 …argspec: "args=[\'map_func\', \'batch_size\', \'num_parallel_batches\', \'drop_remainder\', \'num_…
157 …argspec: "args=[\'map_func\', \'batch_size\', \'num_parallel_batches\', \'drop_remainder\', \'num_…
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | map_and_batch_test.py | 56 def testMapAndBatch(self, num_parallel_calls, num_parallel_batches, argument
75 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_ExperimentalNumaMapAndBatchDataset.pbtxt | 52 to `batch_size * num_parallel_batches` copies of `f` in parallel.
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/ |
| D | imagenet_input.py | 194 num_parallel_batches=self.num_cores,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.experimental.pbtxt | 153 …argspec: "args=[\'map_func\', \'batch_size\', \'num_parallel_batches\', \'drop_remainder\', \'num_…
|