| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | interleave_test.py | 40 def _interleave(lists, cycle_length, block_length, num_parallel_calls=None): argument
187 num_parallel_calls): argument
217 num_parallel_calls): argument
281 block_length, num_parallel_calls): argument
372 def test(self, verify_fn, cycle_length, block_length, num_parallel_calls): argument
390 def testNested(self, verify_fn, num_parallel_calls): argument
|
| D | map_test.py | 205 num_parallel_calls, buffer_size): argument
228 def testParallelMapDataset(self, apply_map, num_parallel_calls, buffer_size): argument
258 def testParallelMapDatasetMultiThreaded(self, apply_map, num_parallel_calls, argument
1035 def testNoInterOpParallelism(self, apply_map, num_parallel_calls): argument
1058 def testShortCircuit(self, apply_map, structure, fn, num_parallel_calls): argument
1072 def testShortCircuitCapturedInput(self, apply_map, num_parallel_calls): argument
1087 def testPreserveCardinality(self, num_parallel_calls): argument
1115 def testMapCancellation(self, apply_map, num_parallel_calls): argument
1339 def testCore(self, verify_fn, num_parallel_calls): argument
1362 def testSaveStatefulFunction(self, num_parallel_calls): argument
[all …]
|
| D | memory_cleanup_test.py | 165 def testMap(self, num_parallel_calls): argument
180 def testMapLegacy(self, num_parallel_calls): argument
195 def testInterleave(self, num_parallel_calls): argument
|
| D | batch_test.py | 58 def testBasic(self, count, batch_size, drop_remainder, num_parallel_calls): argument
|
| /external/tensorflow/tensorflow/python/data/experimental/benchmarks/ |
| D | parameter_value_benchmark.py | 47 def _benchmark_map(self, num_parallel_calls, buffer_size): argument
84 def _benchmark_map_and_batch(self, num_parallel_calls, buffer_size): argument
125 def _benchmark_interleave(self, num_parallel_calls, buffer_size): argument
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | parsing_ops.py | 34 def __init__(self, input_dataset, features, num_parallel_calls, argument
110 def parse_example_dataset(features, num_parallel_calls=1, deterministic=None): argument
|
| D | batching.py | 157 num_parallel_calls=None): argument
217 num_parallel_calls=None): argument
330 def __init__(self, input_dataset, map_func, batch_size, num_parallel_calls, argument
|
| /external/tensorflow/tensorflow/python/data/benchmarks/ |
| D | interleave_benchmark.py | 67 cycle_length, num_parallel_calls): argument
89 num_parallel_calls=None): argument
108 num_parallel_calls=None, argument
|
| D | map_benchmark.py | 154 def _benchmark_nested_parallel_map(self, cycle_length, num_parallel_calls): argument
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | parallel_map_dataset_op.cc | 75 int64_t num_parallel_calls, const DataTypeVector& output_types, in Dataset()
148 Node* num_parallel_calls = nullptr; in AsGraphDefInternal() local
514 int64_t num_parallel_calls = num_parallel_calls_->value; in RunnerThread() local
578 int num_parallel_calls; in StatsThread() local
706 int64_t num_parallel_calls; in MakeDataset() local
|
| D | parallel_batch_dataset_op.cc | 71 Dataset(OpKernelContext* ctx, int64_t batch_size, int64_t num_parallel_calls, in Dataset()
161 Node* num_parallel_calls = nullptr; in AsGraphDefInternal() local
451 int64_t num_parallel_calls = num_parallel_calls_->value; in RunnerThread() local
644 int64_t num_parallel_calls = 0; in MakeDataset() local
|
| D | parallel_batch_dataset_op_test.cc | 30 int64_t num_parallel_calls, bool drop_remainder, in ParallelBatchDatasetParams()
51 Tensor num_parallel_calls = in GetInputTensors() local
|
| D | parallel_map_dataset_op_test.cc | 29 int num_parallel_calls, FunctionDefHelper::AttrValueWrapper func, in ParallelMapDatasetParams()
|
| D | parallel_interleave_dataset_op_test.cc | 30 int64_t num_parallel_calls, FunctionDefHelper::AttrValueWrapper func, in ParallelInterleaveDatasetParams()
|
| D | parallel_interleave_dataset_op.cc | 162 int64_t prefetch_input_elements, int64_t num_parallel_calls, in Dataset()
1652 int64_t num_parallel_calls = 0; in MakeDataset() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_parallelization.cc | 47 auto* num_parallel_calls = graph_utils::AddScalarConstNode( in MakeParallelMap() local
|
| D | batch_parallelization.cc | 47 auto* num_parallel_calls = in MakeParallelBatch() local
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | map_and_batch_dataset_op.cc | 85 int64_t num_parallel_calls, bool drop_remainder, in Dataset()
542 int64_t num_parallel_calls = num_parallel_calls_->value; in RunnerThread() local
722 int64_t num_parallel_calls = 0; in MakeDataset() local
|
| D | parse_example_dataset_op.cc | 121 int64_t num_parallel_calls = 0; in MakeDataset() local
216 example::FastParseExampleConfig config, int32_t num_parallel_calls, in Dataset()
758 int64_t num_parallel_calls = num_parallel_calls_->value; in RunnerThread() local
825 int num_parallel_calls; in StatsThread() local
|
| D | map_and_batch_dataset_op_test.cc | 28 int64_t batch_size, int64_t num_parallel_calls, bool drop_remainder, in MapAndBatchDatasetParams()
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 1530 num_parallel_calls=None, argument
1715 def map(self, map_func, num_parallel_calls=None, deterministic=None): argument
1915 num_parallel_calls=None, argument
3385 num_parallel_calls=None, argument
3402 def map(self, map_func, num_parallel_calls=None, deterministic=None): argument
3418 num_parallel_calls=None, argument
3474 num_parallel_calls=None, argument
4584 num_parallel_calls, deterministic): argument
4971 num_parallel_calls, argument
5093 num_parallel_calls, 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
|