| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | map_fn_test.py | 31 from tensorflow.python.ops import map_fn
58 r = map_fn.map_fn(
65 dtype = map_fn.map_fn(lambda x: constant_op.constant(""),
73 map_fn.map_fn(
83 map_fn.map_fn(lambda x: x, [1, 2])
85 map_fn.map_fn(lambda x: x, 1)
105 r = map_fn.map_fn(double_scoped, elems)
115 r = map_fn.map_fn(double_scoped, elems)
124 y = map_fn.map_fn(
134 r = map_fn.map_fn(
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_vectorization_test.cc | 59 FunctionDef* map_fn = graph->graph()->mutable_library()->add_function(); in AddMapFn() local
60 *map_fn = FunctionDefHelper::Create( in AddMapFn()
68 return map_fn; in AddMapFn()
72 const string& map_fn, int num_parallel_calls = 0) { in AddMapNode() argument
81 {{kAttrNameF, FunctionDefHelper::FunctionRef(map_fn)}, in AddMapNode()
93 {{kAttrNameF, FunctionDefHelper::FunctionRef(map_fn)}, in AddMapNode()
246 auto map_fn = AddMapFn(&graph); in TEST_P() local
248 map_fn->signature().name(), num_parallel_calls); in TEST_P()
285 const string& input_dataset, const string& map_fn, in AddMapAndBatchNode() argument
300 {{kAttrNameF, FunctionDefHelper::FunctionRef(map_fn)}, in AddMapAndBatchNode()
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | map_vectorization_test.py | 326 map_fn, argument
351 dataset = dataset.map(map_fn, num_parallel_calls)
371 def testOptimization(self, map_fn, base_dataset_factory, num_parallel_calls): argument
373 unoptimized, optimized = self._get_test_datasets(base_dataset, map_fn,
379 def map_fn(x): function
387 _, optimized = self._get_test_datasets(base_dataset, map_fn)
396 def map_fn(x): function
402 base_dataset, map_fn, expect_optimized=True)
410 def map_fn(x): function
421 dataset = dataset.apply(batching.map_and_batch(map_fn, 100))
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | flat_map_dataset_serialization_test.py | 41 def map_fn(x): function
44 return dataset_ops.Dataset.range(start, start + 5 * 5, 5).flat_map(map_fn)
54 def map_fn(y): function
57 return dataset_ops.Dataset.range(100).map(map_fn)
67 def map_fn(x): function
75 return dataset_ops.Dataset.range(100).flat_map(map_fn)
95 def map_fn(x): function
99 return dataset_ops.Dataset.range(100).map(map_fn)
|
| D | choose_fastest_dataset_serialization_test.py | 35 map_fn = lambda x: x * 2 function
37 dataset.map(map_fn).batch(batch_size),
38 dataset.batch(batch_size).map(map_fn)
|
| D | map_and_batch_dataset_serialization_test.py | 91 def map_fn(i): function
96 batching.map_and_batch(map_fn, 5))
|
| /external/tensorflow/tensorflow/python/data/benchmarks/ |
| D | map_benchmark.py | 30 def benchmark_helper(chain_length, map_fn, use_inter_op_parallelism, label): argument
34 dataset, map_fn, use_inter_op_parallelism=use_inter_op_parallelism)
49 def benchmark_helper(fan_out, map_fn, use_inter_op_parallelism, label): argument
53 dataset, map_fn, use_inter_op_parallelism=use_inter_op_parallelism)
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_map_fn_op_test.py | 166 output = ragged_map_ops.map_fn(
183 output = ragged_map_ops.map_fn(
205 output = ragged_map_ops.map_fn(
232 output = ragged_map_ops.map_fn(
251 out = ragged_map_ops.map_fn(
266 _ = ragged_map_ops.map_fn(
277 _ = ragged_map_ops.map_fn(
290 id_t2 = ragged_map_ops.map_fn(
|
| /external/tensorflow/tensorflow/python/data/experimental/benchmarks/ |
| D | map_defun_benchmark.py | 28 from tensorflow.python.ops import map_fn
66 map_fn_op = map_fn.map_fn(fn, base)
|
| D | map_vectorization_benchmark.py | 115 def _compare(self, input_dataset, map_fn, batch_size, input_size, str_id): argument
119 unoptimized_dataset = input_dataset.map(map_fn).batch(batch_size)
185 def _benchmark_helper(self, map_fn, str_id, base_dataset_factory=None): argument
197 self._compare(base_dataset, map_fn, batch_size, input_size, str_id)
|
| /external/tensorflow/tensorflow/contrib/gan/python/eval/python/ |
| D | classifier_metrics_impl.py | 46 from tensorflow.python.ops import map_fn
349 logits = map_fn.map_fn(
508 return map_fn.map_fn(fn=classifier_fn,
898 return map_fn.map_fn(
1103 ests = map_fn.map_fn(
|
| D | summaries_impl.py | 26 from tensorflow.python.ops import map_fn
267 map_fn.map_fn(
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_ops.py | 29 from tensorflow.python.ops import map_fn
598 positive_axis = map_fn.map_fn(
605 perm_after = map_fn.map_fn(
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | filter_test_base.py | 26 from tensorflow.python.ops import map_fn
94 squared_xs = map_fn.map_fn(lambda x: x * x, xs)
|
| D | map_test.py | 44 from tensorflow.python.ops import map_fn
77 def map_fn(x): function
83 map_fn, num_parallel_calls).with_options(options)
500 row).map(lambda elems: map_fn.map_fn(lambda x: x * x, elems))
564 row).map(lambda elems: map_fn.map_fn(
606 row).map(lambda elems: map_fn.map_fn(
933 def testShortCircuit(self, structure, map_fn, num_parallel_calls): argument
935 map_fn, num_parallel_calls=num_parallel_calls)
939 expected = map_fn(*self.evaluate(self.structuredElement(structure)))
941 expected = map_fn(self.evaluate(self.structuredElement(structure)))
|
| D | reduce_test.py | 117 def map_fn(i): function
122 ds = dataset_ops.Dataset.range(1, i + 1).map(map_fn)
123 result = ds.reduce(map_fn(0), reduce_fn)
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | multinomial.py | 26 from tensorflow.python.ops import map_fn
269 x = map_fn.map_fn(
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | tf_utils.py | 145 def map_structure_with_atomic(is_atomic_fn, map_fn, nested): argument
162 return map_fn(nested)
173 map_structure_with_atomic(is_atomic_fn, map_fn, ele) for ele in values
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | map_and_batch_test.py | 438 def testShortCircuit(self, structure, map_fn, num_parallel_calls): argument
440 batching.map_and_batch(map_fn, batch_size=10))
444 expected = map_fn(
447 expected = map_fn(
465 def map_fn(x): function
473 batching.map_and_batch(map_fn, batch_size=10))
|
| D | parallel_interleave_test.py | 72 def map_fn(x): function
78 return dataset.map(map_fn)
641 def map_fn(x): function
652 map_fn).repeat(self.repeat_count).apply(
724 map_fn = lambda x: dataset_ops.Dataset.range(20 * x, 20 * (x + 1)) function
727 map_fn,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ExperimentalMapAndBatchDataset.pbtxt | 28 A scalar representing the maximum number of parallel invocations of the `map_fn`
29 function. Applying the `map_fn` on consecutive input elements in parallel has
|
| D | api_def_ExperimentalNumaMapAndBatchDataset.pbtxt | 28 A scalar representing the maximum number of parallel invocations of the `map_fn`
29 function. Applying the `map_fn` on consecutive input elements in parallel has
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | traverse_test.py | 91 def map_fn(ds): function
99 ds2 = ds2.flat_map(map_fn(ds1))
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | while_test.py | 33 from tensorflow.python.ops import map_fn
237 r = map_fn.map_fn(lambda x: math_ops.multiply(math_ops.add(x, 3), 2),
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/ |
| D | __init__.py | 110 map_fn = _ops.map_fn variable
|