Home
last modified time | relevance | path

Searched refs:map_fn (Results 1 – 25 of 91) sorted by relevance

1234

/external/tensorflow/tensorflow/python/kernel_tests/control_flow/
Dmap_fn_test.py30 from tensorflow.python.ops import map_fn
60 r = map_fn.map_fn(
67 dtype = map_fn.map_fn(lambda x: constant_op.constant(""),
78 result = map_fn.map_fn(lambda x: x, st)
87 result = map_fn.map_fn(
102 result = map_fn.map_fn(loop_fn, v)
109 map_fn.map_fn(lambda x: x, [1, 2])
111 map_fn.map_fn(lambda x: x, 1)
131 r = map_fn.map_fn(double_scoped, elems)
141 r = map_fn.map_fn(double_scoped, elems)
[all …]
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dmake_deterministic_test.py65 def map_fn(x): function
71 return dataset_ops.Dataset.range(2).map(map_fn)
74 map_fn = def_function.function(map_fn)
101 def map_fn(x): function
106 map_fn = def_function.function(map_fn)
109 dataset = dataset.map(map_fn, num_parallel_calls=5)
121 def map_fn(x): function
127 dataset = dataset.map(map_fn, num_parallel_calls=5)
140 def map_fn(x): function
146 map_fn = def_function.function(map_fn)
[all …]
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_map_fn_op_test.py24 from tensorflow.python.ops import map_fn as map_fn_lib
181 output = ragged_map_ops.map_fn(
198 output = ragged_map_ops.map_fn(
220 output = ragged_map_ops.map_fn(
247 output = ragged_map_ops.map_fn(
266 out = ragged_map_ops.map_fn(
281 _ = ragged_map_ops.map_fn(
292 _ = ragged_map_ops.map_fn(
305 id_t2 = ragged_map_ops.map_fn(
314 y = map_fn_lib.map_fn(lambda r: map_fn_lib.map_fn(lambda y: r, r), x)
[all …]
Dragged_map_ops.py27 def map_fn(fn, function
161 return map_fn_lib.map_fn(fn,
Dragged_image_ops.py25 from tensorflow.python.ops import map_fn
86 return map_fn.map_fn_v2(resize_one, images, fn_output_signature=out_spec)
/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2021-153.md1 ## TFSA-2021-153: Heap OOB in nested `tf.map_fn` with `RaggedTensor`s
7 It is possible to nest a `tf.map_fn` within another `tf.map_fn` call. However,
15 t = tf.map_fn(lambda r: tf.map_fn(lambda y: r, r), x)
33 t = tf.map_fn(lambda r: tf.map_fn(lambda y: r, r), x)
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dmap_and_batch_test.py330 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 …]
Dfrom_list_test.py62 combinations.combine(map_fn=[
68 def testDatasetInputs(self, map_fn): argument
69 dss = [dataset_ops.Dataset.range(10).map(map_fn) for _ in range(10)]
73 ds, expected_output=[map_fn(x) for x in list(range(10)) * 10])
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dflat_map_test.py192 def map_fn(x): function
195 return dataset_ops.Dataset.range(start, start + 5 * 5, 5).flat_map(map_fn)
221 def map_fn(y): function
224 return dataset_ops.Dataset.range(100).map(map_fn)
237 def map_fn(x): function
245 return dataset_ops.Dataset.range(100).flat_map(map_fn)
267 def map_fn(x): function
271 return dataset_ops.Dataset.range(100).map(map_fn)
Dreduce_test.py121 def map_fn(i): function
126 ds = dataset_ops.Dataset.range(1, i + 1).map(map_fn)
127 result = ds.reduce(map_fn(0), reduce_fn)
204 def map_fn(x): function
209 return dataset_ops.Dataset.range(10).map(map_fn)
Dfilter_test.py25 from tensorflow.python.ops import map_fn
114 squared_xs = map_fn.map_fn(lambda x: x * x, xs)
/external/tensorflow/tensorflow/python/data/experimental/benchmarks/
Dmap_defun_benchmark.py23 from tensorflow.python.ops import map_fn
59 map_fn_op = map_fn.map_fn(fn, base)
/external/tensorflow/tensorflow/compiler/tests/
Dwhile_test.py30 from tensorflow.python.ops import map_fn
234 r = map_fn.map_fn(lambda x: math_ops.multiply(math_ops.add(x, 3), 2),
248 r = map_fn.map_fn(
/external/tensorflow/tensorflow/compiler/mlir/tfrt/saved_model/
Dsaved_model.cc67 llvm::function_ref<void(const TFRTSavedModelSignatureInfo&)> map_fn) { in MapFunctionSignaturesFromTFSavedModelMLIR() argument
71 module.walk([&symbol_table, map_fn, &status](mlir::func::FuncOp func) { in MapFunctionSignaturesFromTFSavedModelMLIR()
138 map_fn(sig_info); in MapFunctionSignaturesFromTFSavedModelMLIR()
/external/tensorflow/tensorflow/python/kernel_tests/math_ops/
Dcumulative_logsumexp_test.py23 from tensorflow.python.ops import map_fn
94 return map_fn.map_fn(
/external/tensorflow/tensorflow/python/data/benchmarks/
Dmap_benchmark.py23 from tensorflow.python.ops import map_fn
135 return map_fn.map_fn(
/external/tensorflow/tensorflow/python/ops/
Dlinalg_ops.py26 from tensorflow.python.ops import map_fn
733 positive_axis = map_fn.map_fn(
742 perm_after = map_fn.map_fn(
Dmap_fn.py44 def map_fn(fn, function
640 return map_fn(
655 map_fn.__doc__)
/external/tensorflow/tensorflow/python/ops/distributions/
Dmultinomial.py22 from tensorflow.python.ops import map_fn
265 x = map_fn.map_fn(
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_MapAndBatchDataset.pbtxt28 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
Dapi_def_ExperimentalMapAndBatchDataset.pbtxt28 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/experimental/ops/
Dbatching.py446 map_fn = lambda *value: nest.map_structure(to_ragged_variant, value) function
448 map_fn = lambda value: nest.map_structure(to_ragged_variant, value) function
450 self._mapped_dataset = input_dataset.map(map_fn)
/external/perfetto/include/perfetto/ext/base/threading/
Dstream.h64 Stream<FutureReturn<Function, T>> MapFuture(Function map_fn) && { in MapFuture() argument
66 std::move(map_fn)); in MapFuture()
/external/tensorflow/tensorflow/python/data/util/
Dtraverse_test.py92 def map_fn(ds): function
100 ds2 = ds2.flat_map(map_fn(ds1))
/external/tensorflow/tensorflow/python/keras/utils/
Dtf_utils.py106 def map_structure_with_atomic(is_atomic_fn, map_fn, nested): argument
123 return map_fn(nested)
136 map_structure_with_atomic(is_atomic_fn, map_fn, ele) for ele in values

1234