| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | parsing_ops.py | 33 def __init__(self, input_dataset, features, num_parallel_calls): argument
100 def parse_example_dataset(features, num_parallel_calls=1): argument
|
| D | batching.py | 568 def __init__(self, input_dataset, map_func, batch_size, num_parallel_calls, argument
618 num_parallel_calls=None): argument
678 num_parallel_calls=None): argument
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | interleave_test.py | 143 num_parallel_calls): argument
171 num_parallel_calls): argument
223 block_length, num_parallel_calls): argument
|
| D | map_test.py | 55 def _make_coordinated_sloppy_dataset(num_elements, num_parallel_calls): argument
155 def _buildParallelMapDataset(self, components, count, num_parallel_calls, argument
175 def do_test(num_parallel_calls, output_buffer_size): argument
203 def do_test(num_parallel_calls, output_buffer_size): argument
933 def testShortCircuit(self, structure, map_fn, num_parallel_calls): argument
948 def testShortCircuitCapturedInput(self, num_parallel_calls): argument
965 def testSloppyInterleaveInOrder(self, num_elements, num_parallel_calls): argument
985 def testSloppyInterleaveOutOfOrder(self, num_elements, num_parallel_calls): argument
1010 def testPreserveCardinality(self, num_parallel_calls): argument
|
| D | dataset_test.py | 71 def make_interleave_fn(dataset, num_parallel_calls=None): argument
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | interleave_dataset_serialization_test.py | 35 num_parallel_calls): argument
53 num_parallel_calls): argument
|
| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| D | parsing_ops.py | 26 def parse_example_dataset(features, num_parallel_calls=1): argument
|
| D | batching.py | 236 num_parallel_calls=None): argument
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | parallel_map_dataset_op.cc | 56 int32 num_parallel_calls; in MakeDataset() local
83 const NameAttrList& func, int32 num_parallel_calls, in Dataset()
163 Node* num_parallel_calls = nullptr; in AsGraphDefInternal() local
|
| D | parallel_map_iterator.cc | 44 int32 num_parallel_calls; member
280 int64 num_parallel_calls = num_parallel_calls_->value; in RunnerThread() local
410 int32 num_parallel_calls, bool sloppy, bool preserve_cardinality) { in NewParallelMapIterator()
|
| D | parallel_interleave_dataset_op.cc | 77 int64 num_parallel_calls; in MakeDataset() local
110 int64 block_length, int64 num_parallel_calls, bool sloppy, in Dataset()
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/ |
| D | cifar_input.py | 38 num_parallel_calls=12, argument
|
| D | imagenet_input.py | 77 num_parallel_calls=64, argument
|
| /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
438 def testShortCircuit(self, structure, map_fn, num_parallel_calls): argument
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_parallelization.cc | 47 auto* num_parallel_calls = graph_utils::AddScalarConstNode(kAutotune, graph); in MakeParallelMap() local
|
| D | map_vectorization_test.cc | 72 const string& map_fn, int num_parallel_calls = 0) { in AddMapNode()
240 int num_parallel_calls = std::get<0>(GetParam()); in TEST_P() local
287 int64 num_parallel_calls = 12) { in AddMapAndBatchNode()
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | map_vectorization_test.py | 327 num_parallel_calls=None, argument
371 def testOptimization(self, map_fn, base_dataset_factory, num_parallel_calls): argument
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | parse_example_dataset_op.cc | 75 int64 num_parallel_calls; in MakeDataset() local
160 example::FastParseExampleConfig config, int32 num_parallel_calls, in Dataset()
|
| D | map_and_batch_dataset_op.cc | 76 int64 num_parallel_calls; in MakeDataset() local
148 int64 num_parallel_calls, bool drop_remainder, in Dataset()
607 int64 num_parallel_calls = num_parallel_calls_->value; in RunnerThread() local
|
| D | numa_map_and_batch_dataset_op.cc | 75 int64 num_parallel_calls; in MakeDataset() local
102 int64 num_parallel_calls, bool drop_remainder, in Dataset()
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 929 def map(self, map_func, num_parallel_calls=None): argument
1061 num_parallel_calls=None): argument
1661 def map(self, map_func, num_parallel_calls=None): argument
1671 def map_with_legacy_function(self, map_func, num_parallel_calls=None): argument
1716 num_parallel_calls=None): argument
3032 num_parallel_calls, argument
3142 num_parallel_calls): argument
|