Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dmap_vectorization_test.py204 num_parallel_calls=None, argument
246 def _testOptimization(self, map_fn, dataset_factory, num_parallel_calls): argument
255 def testBasic(self, num_parallel_calls): argument
264 def testBroadcast(self, num_parallel_calls): argument
274 def testCast(self, num_parallel_calls): argument
283 def testConst(self, num_parallel_calls): argument
292 def testCycle(self, num_parallel_calls): argument
305 def testReshape(self, num_parallel_calls): argument
314 def testTranspose(self, num_parallel_calls): argument
323 def testUnstack(self, num_parallel_calls): argument
[all …]
/external/tensorflow/tensorflow/python/data/experimental/benchmarks/
Dparameter_value_benchmark.py30 def _benchmark_map(self, num_parallel_calls, buffer_size): argument
62 def _benchmark_map_and_batch(self, num_parallel_calls, buffer_size): argument
99 def _benchmark_interleave(self, num_parallel_calls, buffer_size): argument
Dparallel_interleave_benchmark.py68 cycle_length, num_parallel_calls): argument
90 num_parallel_calls=None): argument
107 num_parallel_calls=None, argument
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dinterleave_test.py38 def _interleave(lists, cycle_length, block_length, num_parallel_calls=None): argument
185 num_parallel_calls): argument
215 num_parallel_calls): argument
279 block_length, num_parallel_calls): argument
Dmemory_cleanup_test.py174 def testMap(self, num_parallel_calls): argument
189 def testMapLegacy(self, num_parallel_calls): argument
204 def testInterleave(self, num_parallel_calls): argument
Dmap_test.py122 num_parallel_calls): argument
231 num_parallel_calls, buffer_size): argument
254 def testParallelMapDataset(self, apply_map, num_parallel_calls, buffer_size): argument
284 def testParallelMapDatasetMultiThreaded(self, apply_map, num_parallel_calls, argument
1053 def testNoInterOpParallelism(self, apply_map, num_parallel_calls): argument
1076 def testShortCircuit(self, apply_map, structure, fn, num_parallel_calls): argument
1090 def testShortCircuitCapturedInput(self, apply_map, num_parallel_calls): argument
1110 num_parallel_calls): argument
1132 num_parallel_calls): argument
1158 def testPreserveCardinality(self, num_parallel_calls): argument
[all …]
Dbatch_test.py50 def testBasic(self, count, batch_size, drop_remainder, num_parallel_calls): argument
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dparsing_ops.py34 def __init__(self, input_dataset, features, num_parallel_calls, argument
110 def parse_example_dataset(features, num_parallel_calls=1, deterministic=None): argument
Dbatching.py157 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/experimental/kernel_tests/serialization/
Dinterleave_dataset_serialization_test.py37 num_parallel_calls): argument
52 num_parallel_calls): argument
/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_batch_dataset_op.cc66 Dataset(OpKernelContext* ctx, int64 batch_size, int64 num_parallel_calls, in Dataset()
152 Node* num_parallel_calls = nullptr; in AsGraphDefInternal() local
409 int64 num_parallel_calls = num_parallel_calls_->value; in RunnerThread() local
535 int64 num_parallel_calls = 0; in MakeDataset() local
Dparallel_map_dataset_op.cc75 int64 num_parallel_calls, const DataTypeVector& output_types, in Dataset()
148 Node* num_parallel_calls = nullptr; in AsGraphDefInternal() local
513 int64 num_parallel_calls = num_parallel_calls_->value; in RunnerThread() local
577 int num_parallel_calls; in StatsThread() local
706 int64 num_parallel_calls; in MakeDataset() local
Dparallel_batch_dataset_op_test.cc30 int64 num_parallel_calls, bool drop_remainder, in ParallelBatchDatasetParams()
48 Tensor num_parallel_calls = in GetInputTensors() local
Dparallel_map_dataset_op_test.cc29 int num_parallel_calls, FunctionDefHelper::AttrValueWrapper func, in ParallelMapDatasetParams()
Dparallel_interleave_dataset_op_test.cc29 int64 prefetch_input_elements, int64 num_parallel_calls, in ParallelInterleaveDatasetParams()
Dparallel_interleave_dataset_op.cc164 int64 prefetch_input_elements, int64 num_parallel_calls, in Dataset()
1626 int64 num_parallel_calls = 0; in MakeDataset() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_parallelization.cc47 auto* num_parallel_calls = graph_utils::AddScalarConstNode( in MakeParallelMap() local
Dbatch_parallelization.cc48 auto* num_parallel_calls = in MakeParallelBatch() local
Dmap_vectorization_test.cc85 const string& map_fn, int num_parallel_calls = 0) { in AddMapNode()
287 int num_parallel_calls = std::get<0>(GetParam()); in TEST_P() local
344 int64 num_parallel_calls = 12) { in AddMapAndBatchNode()
/external/tensorflow/tensorflow/python/data/benchmarks/
Dmap_benchmark.py119 def _benchmark_nested_parallel_map(self, cycle_length, num_parallel_calls): argument
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dmap_and_batch_dataset_op.cc85 int64 num_parallel_calls, bool drop_remainder, in Dataset()
536 int64 num_parallel_calls = num_parallel_calls_->value; in RunnerThread() local
708 int64 num_parallel_calls = 0; in MakeDataset() local
Dparse_example_dataset_op.cc121 int64 num_parallel_calls = 0; in MakeDataset() local
216 example::FastParseExampleConfig config, int32 num_parallel_calls, in Dataset()
749 int64 num_parallel_calls = num_parallel_calls_->value; in RunnerThread() local
816 int num_parallel_calls; in StatsThread() local
Dmap_and_batch_dataset_op_test.cc28 int64 batch_size, int64 num_parallel_calls, bool drop_remainder, in MapAndBatchDatasetParams()
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py1509 def batch(self, batch_size, drop_remainder=False, num_parallel_calls=None): argument
1678 def map(self, map_func, num_parallel_calls=None, deterministic=None): argument
1863 num_parallel_calls=None, argument
2635 def batch(self, batch_size, drop_remainder=False, num_parallel_calls=None): argument
2651 def map(self, map_func, num_parallel_calls=None, deterministic=None): argument
2667 num_parallel_calls=None, argument
2723 num_parallel_calls=None, argument
3994 num_parallel_calls): argument
4344 num_parallel_calls, argument
4466 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