Home
last modified time | relevance | path

Searched defs:num_parallel_calls (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/python/data/kernel_tests/
Dinterleave_test.py36 def _interleave(lists, cycle_length, block_length, num_parallel_calls=None): argument
189 num_parallel_calls): argument
219 num_parallel_calls): argument
283 block_length, num_parallel_calls): argument
365 def testName(self, num_parallel_calls): argument
377 def testMapFuncMustReturnDataset(self, num_parallel_calls): argument
399 def test(self, verify_fn, cycle_length, block_length, num_parallel_calls): argument
417 def testNested(self, verify_fn, num_parallel_calls): argument
459 def test(self, verify_fn, cycle_length, block_length, num_parallel_calls): argument
477 def testNested(self, verify_fn, num_parallel_calls): argument
Dmap_test.py203 num_parallel_calls, buffer_size): argument
226 def testParallelMapDataset(self, apply_map, num_parallel_calls, buffer_size): argument
256 def testParallelMapDatasetMultiThreaded(self, apply_map, num_parallel_calls, argument
1033 def testNoInterOpParallelism(self, apply_map, num_parallel_calls): argument
1056 def testShortCircuit(self, apply_map, structure, fn, num_parallel_calls): argument
1070 def testShortCircuitCapturedInput(self, apply_map, num_parallel_calls): argument
1085 def testPreserveCardinality(self, num_parallel_calls): argument
1113 def testMapCancellation(self, apply_map, num_parallel_calls): argument
1332 def testName(self, num_parallel_calls): argument
1345 def testCore(self, verify_fn, num_parallel_calls): argument
[all …]
Dmemory_cleanup_test.py160 def testMap(self, num_parallel_calls): argument
175 def testMapLegacy(self, num_parallel_calls): argument
190 def testInterleave(self, num_parallel_calls): argument
Dbatch_test.py55 def testBasic(self, count, batch_size, drop_remainder, num_parallel_calls): argument
296 def testName(self, num_parallel_calls): argument
/external/tensorflow/tensorflow/python/data/experimental/benchmarks/
Dparameter_value_benchmark.py43 def _benchmark_map(self, num_parallel_calls, buffer_size): argument
80 def _benchmark_map_and_batch(self, num_parallel_calls, buffer_size): argument
121 def _benchmark_interleave(self, num_parallel_calls, buffer_size): argument
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dparsing_ops.py30 def __init__(self, input_dataset, features, num_parallel_calls, argument
107 def parse_example_dataset(features, num_parallel_calls=1, deterministic=None): argument
Dbatching.py154 num_parallel_calls=None): argument
218 num_parallel_calls=None): argument
335 def __init__(self, input_dataset, map_func, batch_size, num_parallel_calls, argument
/external/tensorflow/tensorflow/python/data/benchmarks/
Dinterleave_benchmark.py63 cycle_length, num_parallel_calls): argument
85 num_parallel_calls=None): argument
104 num_parallel_calls=None, argument
Dmap_benchmark.py150 def _benchmark_nested_parallel_map(self, cycle_length, num_parallel_calls): argument
/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_map_dataset_op.cc78 int64_t num_parallel_calls, const DataTypeVector& output_types, in Dataset()
88 int64_t num_parallel_calls, const DataTypeVector& output_types, in Dataset()
182 Node* num_parallel_calls = nullptr; in AsGraphDefInternal() local
553 int64_t num_parallel_calls = num_parallel_calls_->value; in RunnerThread() local
617 int num_parallel_calls; in StatsThread() local
754 int64_t num_parallel_calls; in MakeDataset() local
Dparallel_batch_dataset_op.cc71 Dataset(OpKernelContext* ctx, int64_t batch_size, int64_t num_parallel_calls, in Dataset()
161 Node* num_parallel_calls = nullptr; in AsGraphDefInternal() local
456 int64_t num_parallel_calls = num_parallel_calls_->value; in RunnerThread() local
656 int64_t num_parallel_calls = 0; in MakeDataset() local
Dparallel_filter_dataset_op.cc64 int64_t num_parallel_calls, DeterminismPolicy deterministic, in Dataset()
114 Node* num_parallel_calls = nullptr; in AsGraphDefInternal() local
428 int64_t num_parallel_calls = num_parallel_calls_->value; in RunnerThread() local
645 int64_t num_parallel_calls; in MakeDataset() local
Dparallel_batch_dataset_op_test.cc30 int64_t num_parallel_calls, bool drop_remainder, in ParallelBatchDatasetParams()
51 Tensor num_parallel_calls = in GetInputTensors() local
Dparallel_filter_dataset_op_test.cc33 int num_parallel_calls, const std::string& deterministic, in ParallelFilterDatasetParams()
Dparallel_map_dataset_op_test.cc29 int num_parallel_calls, FunctionDefHelper::AttrValueWrapper func, in ParallelMapDatasetParams()
Dparallel_interleave_dataset_op.cc136 int64_t num_parallel_calls) { in ComputeCycleLength()
201 int64_t prefetch_input_elements, int64_t num_parallel_calls, in Dataset()
1735 int64_t num_parallel_calls = 0; in MakeDataset() local
Dparallel_interleave_dataset_op_test.cc35 int64_t num_parallel_calls, FunctionDefHelper::AttrValueWrapper func, in ParallelInterleaveDatasetParams()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfilter_parallelization.cc47 auto* num_parallel_calls = graph_utils::AddScalarConstNode( in MakeParallelFilter() local
Dbatch_parallelization.cc47 auto* num_parallel_calls = in MakeParallelBatch() local
Dmap_parallelization.cc47 auto* num_parallel_calls = graph_utils::AddScalarConstNode( in MakeParallelMap() local
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dmap_and_batch_dataset_op.cc85 int64_t num_parallel_calls, bool drop_remainder, in Dataset()
547 int64_t num_parallel_calls = num_parallel_calls_->value; in RunnerThread() local
734 int64_t num_parallel_calls = 0; in MakeDataset() local
Dparse_example_dataset_op.cc121 int64_t num_parallel_calls = 0; in MakeDataset() local
216 example::FastParseExampleConfig config, int32_t num_parallel_calls, in Dataset()
760 int64_t num_parallel_calls = num_parallel_calls_->value; in RunnerThread() local
827 int num_parallel_calls; in StatsThread() local
Dmap_and_batch_dataset_op_test.cc28 int64_t batch_size, int64_t num_parallel_calls, bool drop_remainder, in MapAndBatchDatasetParams()
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py1835 num_parallel_calls=None, argument
2038 num_parallel_calls=None, argument
2242 num_parallel_calls=None, argument
3976 num_parallel_calls=None, argument
4005 num_parallel_calls=None, argument
4023 num_parallel_calls=None, argument
4080 num_parallel_calls=None, argument
5137 num_parallel_calls, argument
5438 num_parallel_calls, argument
5568 num_parallel_calls, argument
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dmap_and_batch_test.py52 def testMapAndBatch(self, num_parallel_calls, num_parallel_batches): argument