| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | map_fn_test.py | 32 from tensorflow.python.ops import map_fn
61 r = map_fn.map_fn(
68 dtype = map_fn.map_fn(lambda x: constant_op.constant(""),
79 result = map_fn.map_fn(lambda x: x, st)
88 result = map_fn.map_fn(
98 map_fn.map_fn(lambda x: x, [1, 2])
100 map_fn.map_fn(lambda x: x, 1)
120 r = map_fn.map_fn(double_scoped, elems)
130 r = map_fn.map_fn(double_scoped, elems)
139 y = map_fn.map_fn(
[all …]
|
| D | cumulative_logsumexp_test.py | 27 from tensorflow.python.ops import map_fn
98 return map_fn.map_fn(
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | map_vectorization_test.py | 55 return x + combinations.combine(map_fn=combinations.NamedObject(name, fn))
203 map_fn, argument
230 dataset = dataset.map(map_fn, num_parallel_calls)
246 def _testOptimization(self, map_fn, dataset_factory, num_parallel_calls): argument
248 unoptimized, optimized = self._get_test_datasets(dataset, map_fn,
258 map_fn = lambda x: (x, x + 1) function
259 self._testOptimization(map_fn, dataset_factory, num_parallel_calls)
268 map_fn = lambda x: x + value function
269 self._testOptimization(map_fn, dataset_factory, num_parallel_calls)
277 map_fn = lambda x: math_ops.cast(x, dtypes.float64) function
[all …]
|
| D | hoist_random_uniform_test.py | 60 name, map_fn, should_optimize = y
62 map_fn=combinations.NamedObject(name, map_fn),
89 def testHoistFunction(self, map_fn, should_optimize): argument
92 ["Zip[0]", "Map"] if should_optimize else ["Map"])).map(map_fn)
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | preprocessing_stage.py | 76 def map_fn(x): function
93 current_layer_data = data.map(map_fn)
95 current_layer_data = map_fn(data)
210 def map_fn(*x): function
214 def map_fn(*x): function
217 return map_fn
235 map_fn = build_map_fn(node, args, kwargs)
236 outputs = args.map(map_fn)
270 def map_fn(*x, j=i): function
273 ds_unzipped.append(ds.map(map_fn))
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | flat_map_dataset_serialization_test.py | 47 def map_fn(x): function
50 return dataset_ops.Dataset.range(start, start + 5 * 5, 5).flat_map(map_fn)
61 def map_fn(y): function
64 return dataset_ops.Dataset.range(100).map(map_fn)
75 def map_fn(x): function
83 return dataset_ops.Dataset.range(100).flat_map(map_fn)
105 def map_fn(x): function
109 return dataset_ops.Dataset.range(100).map(map_fn)
|
| D | choose_fastest_dataset_serialization_test.py | 41 map_fn = lambda x: x * 2 function
43 dataset.map(map_fn).batch(batch_size),
44 dataset.batch(batch_size).map(map_fn)
|
| D | map_and_batch_dataset_serialization_test.py | 95 def map_fn(i): function
100 batching.map_and_batch(map_fn, 5))
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_vectorization_test.cc | 72 FunctionDef* map_fn = graph->graph()->mutable_library()->add_function(); in AddMapFn() local
73 *map_fn = FunctionDefHelper::Create( in AddMapFn()
81 return map_fn; in AddMapFn()
85 const string& map_fn, int num_parallel_calls = 0) { in AddMapNode() argument
94 {{kAttrNameF, FunctionDefHelper::FunctionRef(map_fn)}, in AddMapNode()
106 {{kAttrNameF, FunctionDefHelper::FunctionRef(map_fn)}, in AddMapNode()
294 auto map_fn = AddMapFn(&graph); in TEST_P() local
296 map_fn->signature().name(), num_parallel_calls); in TEST_P()
342 const string& input_dataset, const string& map_fn, in AddMapAndBatchNode() argument
357 {{kAttrNameF, FunctionDefHelper::FunctionRef(map_fn)}, in AddMapAndBatchNode()
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | map_and_batch_test.py | 330 map_fn = lambda x: x function
332 batching.map_and_batch(map_fn, batch_size=10))
334 expected = map_fn(self.evaluate(self.structuredElement(None, shape=[10])))
339 map_fn = lambda x: (x, x) function
341 batching.map_and_batch(map_fn, batch_size=10))
343 expected = map_fn(self.evaluate(self.structuredElement(None, shape=[10])))
348 map_fn = lambda x, y: (y, x) function
351 None)).repeat().apply(batching.map_and_batch(map_fn, batch_size=10))
353 expected = map_fn(
359 map_fn = lambda x, y: x function
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_map_fn_op_test.py | 183 output = ragged_map_ops.map_fn(
200 output = ragged_map_ops.map_fn(
222 output = ragged_map_ops.map_fn(
249 output = ragged_map_ops.map_fn(
268 out = ragged_map_ops.map_fn(
283 _ = ragged_map_ops.map_fn(
294 _ = ragged_map_ops.map_fn(
307 id_t2 = ragged_map_ops.map_fn(
|
| D | ragged_map_ops.py | 30 def map_fn(fn, function
164 return map_fn_lib.map_fn(fn,
|
| /external/tensorflow/tensorflow/python/data/experimental/benchmarks/ |
| D | map_defun_benchmark.py | 27 from tensorflow.python.ops import map_fn
59 map_fn_op = map_fn.map_fn(fn, base)
|
| D | map_vectorization_benchmark.py | 108 def _compare(self, input_dataset, map_fn, batch_size, input_size, str_id): argument
112 unoptimized_dataset = input_dataset.map(map_fn).batch(batch_size)
176 def _benchmark_helper(self, map_fn, str_id, base_dataset_factory=None): argument
188 self._compare(base_dataset, map_fn, batch_size, input_size, str_id)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | while_test.py | 34 from tensorflow.python.ops import map_fn
238 r = map_fn.map_fn(lambda x: math_ops.multiply(math_ops.add(x, 3), 2),
252 r = map_fn.map_fn(
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | reduce_test.py | 125 def map_fn(i): function
130 ds = dataset_ops.Dataset.range(1, i + 1).map(map_fn)
131 result = ds.reduce(map_fn(0), reduce_fn)
208 def map_fn(x): function
213 return dataset_ops.Dataset.range(10).map(map_fn)
|
| D | filter_test.py | 28 from tensorflow.python.ops import map_fn
117 squared_xs = map_fn.map_fn(lambda x: x * x, xs)
|
| /external/tensorflow/tensorflow/python/data/benchmarks/ |
| D | map_benchmark.py | 28 from tensorflow.python.ops import map_fn
109 return map_fn.map_fn(
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_ops.py | 30 from tensorflow.python.ops import map_fn
734 positive_axis = map_fn.map_fn(
743 perm_after = map_fn.map_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/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_MapAndBatchDataset.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/keras/engine/ |
| D | training_integration_test.py | 154 def map_fn(_): function
159 dataset = dataset_ops.DatasetV2.range(4).map(map_fn).batch(batch_size)
177 eval_dataset = dataset_ops.DatasetV2.range(4).map(map_fn).batch(batch_size)
|
| /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/python/data/experimental/ops/ |
| D | batching.py | 441 map_fn = lambda *value: nest.map_structure(to_ragged_variant, value) function
443 map_fn = lambda value: nest.map_structure(to_ragged_variant, value) function
445 self._mapped_dataset = input_dataset.map(map_fn)
|