Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_map_dataset_op.cc56 int32 num_parallel_calls; in MakeDataset() local
58 &num_parallel_calls)); in MakeDataset()
60 ctx, num_parallel_calls > 0 || num_parallel_calls == model::kAutoTune, in MakeDataset()
69 if (num_parallel_calls == model::kAutoTune) { in MakeDataset()
73 *output = new Dataset(ctx, input, func_, num_parallel_calls, output_types_, in MakeDataset()
83 const NameAttrList& func, int32 num_parallel_calls, in Dataset() argument
92 num_parallel_calls_(num_parallel_calls), in Dataset()
163 Node* num_parallel_calls = nullptr; in AsGraphDefInternal() local
165 b->AddScalar(num_parallel_calls_, &num_parallel_calls)); in AsGraphDefInternal()
192 std::make_pair(2, num_parallel_calls)}, // Single tensor inputs. in AsGraphDefInternal()
Dparallel_map_iterator.cc37 int32 num_parallel_calls, bool sloppy, bool preserve_cardinality) in Params()
39 num_parallel_calls(num_parallel_calls), in Params()
44 int32 num_parallel_calls; member
58 params.num_parallel_calls, mu_, cond_var_)), in ParallelMapIterator()
280 int64 num_parallel_calls = num_parallel_calls_->value; in RunnerThread() local
281 return num_calls_ >= num_parallel_calls || in RunnerThread()
282 invocation_results_.size() >= num_parallel_calls; in RunnerThread()
410 int32 num_parallel_calls, bool sloppy, bool preserve_cardinality) { in NewParallelMapIterator() argument
414 num_parallel_calls, sloppy, in NewParallelMapIterator()
Dparallel_interleave_dataset_op.cc77 int64 num_parallel_calls; in MakeDataset() local
79 &num_parallel_calls)); in MakeDataset()
81 ctx, num_parallel_calls > 0 || num_parallel_calls == model::kAutoTune, in MakeDataset()
85 ctx, num_parallel_calls <= cycle_length, in MakeDataset()
94 if (num_parallel_calls == model::kAutoTune) { in MakeDataset()
100 cycle_length, block_length, num_parallel_calls, sloppy_, in MakeDataset()
110 int64 block_length, int64 num_parallel_calls, bool sloppy, in Dataset() argument
119 num_parallel_calls_(num_parallel_calls), in Dataset()
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dinterleave_dataset_serialization_test.py35 num_parallel_calls): argument
40 cycle_length, block_length, num_parallel_calls)
53 num_parallel_calls): argument
59 input_values, cycle_length, block_length, num_parallel_calls),
61 input_values, cycle_length * 2, block_length, num_parallel_calls),
Dparallel_map_dataset_serialization_test.py51 _map_fn, num_parallel_calls=3).repeat(self._num_epochs))
62 _map_fn, num_parallel_calls=3).repeat(self._num_epochs).prefetch(5))
80 _map_fn, num_parallel_calls=2).prefetch(2)
91 num_parallel_calls=2).prefetch(2))
100 lambda x: x + constant_var, num_parallel_calls=2).prefetch(2))
114 defun_fn, num_parallel_calls=2).prefetch(2)
134 defun_fn, num_parallel_calls=2).prefetch(2)
Dmap_and_batch_dataset_serialization_test.py67 num_parallel_calls = 7
79 num_parallel_calls=num_parallel_calls,
Dnuma_map_and_batch_dataset_serialization_test.py69 num_parallel_calls = 7
81 num_parallel_calls=num_parallel_calls,
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dbatching.py568 def __init__(self, input_dataset, map_func, batch_size, num_parallel_calls, argument
581 num_parallel_calls, dtype=dtypes.int64, name="num_parallel_calls")
598 num_parallel_calls=self._num_parallel_calls_t,
618 num_parallel_calls=None): argument
652 if num_parallel_batches is None and num_parallel_calls is None:
653 num_parallel_calls = batch_size
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:
662 num_parallel_calls, drop_remainder,
[all …]
Dparsing_ops.py33 def __init__(self, input_dataset, features, num_parallel_calls): argument
38 self._num_parallel_calls = num_parallel_calls
100 def parse_example_dataset(features, num_parallel_calls=1): argument
132 out_dataset = _ParseExampleDataset(dataset, features, num_parallel_calls)
141 features, x), num_parallel_calls=num_parallel_calls)
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dmap_test.py55 def _make_coordinated_sloppy_dataset(num_elements, num_parallel_calls): argument
83 map_fn, num_parallel_calls).with_options(options)
155 def _buildParallelMapDataset(self, components, count, num_parallel_calls, argument
162 _map_fn, num_parallel_calls=num_parallel_calls).prefetch(
175 def do_test(num_parallel_calls, output_buffer_size): argument
182 self._buildParallelMapDataset(components, 14, num_parallel_calls,
203 def do_test(num_parallel_calls, output_buffer_size): argument
209 self._buildParallelMapDataset(components, 18, num_parallel_calls,
264 num_parallel_calls=2))
275 num_parallel_calls=2))
[all …]
Dinterleave_test.py143 num_parallel_calls): argument
148 cycle_length, block_length, num_parallel_calls)
171 num_parallel_calls): argument
175 num_parallel_calls)
223 block_length, num_parallel_calls): argument
228 cycle_length, block_length, num_parallel_calls)
Ddataset_test.py71 def make_interleave_fn(dataset, num_parallel_calls=None): argument
77 num_parallel_calls=num_parallel_calls)
126 lambda x: x.map(lambda x: x, num_parallel_calls=2),
/external/tensorflow/tensorflow/python/data/experimental/benchmarks/
Dautotune_benchmark.py46 math_ops.matmul, num_parallel_calls=optimization.AUTOTUNE)
84 num_parallel_calls=optimization.AUTOTUNE,
123 num_parallel_calls=optimization.AUTOTUNE)
169 dataset = dataset.map(f1, num_parallel_calls=optimization.AUTOTUNE)
172 num_parallel_calls=optimization.AUTOTUNE,
176 dataset = dataset.map(f2, num_parallel_calls=optimization.AUTOTUNE)
179 num_parallel_calls=optimization.AUTOTUNE,
183 dataset = dataset.map(f2, num_parallel_calls=optimization.AUTOTUNE)
/external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/
Dimagenet_input.py77 num_parallel_calls=64, argument
86 self.num_parallel_calls = num_parallel_calls
173 fetch_dataset, cycle_length=self.num_parallel_calls, sloppy=True))
201 num_parallel_calls=self.num_cores)
220 num_parallel_calls=8)
Dcifar_input.py38 num_parallel_calls=12, argument
101 dataset = dataset.map(_parser, num_parallel_calls=num_parallel_calls)
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_vectorization_test.cc72 const string& map_fn, int num_parallel_calls = 0) { in AddMapNode() argument
74 if (num_parallel_calls) { in AddMapNode()
76 graph_utils::AddScalarConstNode(num_parallel_calls, graph); in AddMapNode()
240 int num_parallel_calls = std::get<0>(GetParam()); in TEST_P() local
248 map_fn->signature().name(), num_parallel_calls); in TEST_P()
259 expected_original_branch.push_back(num_parallel_calls > 0 ? kParallelMapOp in TEST_P()
269 expected_vectorized_branch.push_back(num_parallel_calls > 0 ? kParallelMapOp in TEST_P()
287 int64 num_parallel_calls = 12) { in AddMapAndBatchNode() argument
290 graph_utils::AddScalarConstNode(num_parallel_calls, graph); in AddMapAndBatchNode()
Dmap_parallelization.cc47 auto* num_parallel_calls = graph_utils::AddScalarConstNode(kAutotune, graph); in MakeParallelMap() local
48 parallel_map.add_input(num_parallel_calls->name()); in MakeParallelMap()
/external/tensorflow/tensorflow/contrib/data/python/ops/
Dparsing_ops.py26 def parse_example_dataset(features, num_parallel_calls=1): argument
53 return parsing_ops.parse_example_dataset(features, num_parallel_calls)
Dbatching.py236 num_parallel_calls=None): argument
273 drop_remainder, num_parallel_calls)
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dmap_vectorization_test.py311 x[0] + "Parallel" if num_parallel_calls is not None else x[0],
317 num_parallel_calls
318 } for x in test_cases for num_parallel_calls in (None, 12)]
327 num_parallel_calls=None, argument
347 map_node_name = "Map" if num_parallel_calls is None else "ParallelMap"
351 dataset = dataset.map(map_fn, num_parallel_calls)
371 def testOptimization(self, map_fn, base_dataset_factory, num_parallel_calls): argument
374 num_parallel_calls)
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ParallelMapDataset.pbtxt5 name: "num_parallel_calls"
14 to `num_parallel_calls` copies of `f` in parallel.
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dparse_example_dataset_op.cc75 int64 num_parallel_calls; in MakeDataset() local
77 &num_parallel_calls)); in MakeDataset()
78 OP_REQUIRES(ctx, num_parallel_calls > 0, in MakeDataset()
149 num_parallel_calls, sparse_types_, dense_types_, in MakeDataset()
160 example::FastParseExampleConfig config, int32 num_parallel_calls, in Dataset() argument
173 num_parallel_calls_(num_parallel_calls), in Dataset()
Dmap_and_batch_dataset_op.cc76 int64 num_parallel_calls; in MakeDataset() local
78 &num_parallel_calls)); in MakeDataset()
80 ctx, num_parallel_calls > 0 || num_parallel_calls == model::kAutoTune, in MakeDataset()
133 if (num_parallel_calls == model::kAutoTune) { in MakeDataset()
137 *output = new Dataset(ctx, input, func_, batch_size, num_parallel_calls, in MakeDataset()
148 int64 num_parallel_calls, bool drop_remainder, in Dataset() argument
158 num_parallel_calls_(num_parallel_calls), in Dataset()
607 int64 num_parallel_calls = num_parallel_calls_->value; in RunnerThread() local
608 return num_calls_ >= num_parallel_calls || in RunnerThread()
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py929 def map(self, map_func, num_parallel_calls=None): argument
1021 if num_parallel_calls is None:
1025 self, map_func, num_parallel_calls, preserve_cardinality=True)
1061 num_parallel_calls=None): argument
1134 if num_parallel_calls is None:
1138 block_length, num_parallel_calls)
1661 def map(self, map_func, num_parallel_calls=None): argument
1662 if num_parallel_calls is None:
1668 self, map_func, num_parallel_calls, preserve_cardinality=False))
1671 def map_with_legacy_function(self, map_func, num_parallel_calls=None): argument
[all …]
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_utils_test.py121 num_parallel_calls=1), ValueError),
129 lambda x: x, num_parallel_calls=1)),
174 num_parallel_calls=1), ValueError),
176 lambda x: x, num_parallel_calls=1)),

123