| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | make_deterministic_test.py | 65 def map_fn(x): function
101 def map_fn(x): function
121 def map_fn(x): function
140 def map_fn(x): function
165 def map_fn(x): function
198 def map_fn(x): function
260 def map_fn(x): function
350 def map_fn(x): function
|
| D | optimization_test.py | 52 def map_fn(x): function
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | map_and_batch_test.py | 330 map_fn = lambda x: x function
339 map_fn = lambda x: (x, x) function
348 map_fn = lambda x, y: (y, x) function
359 map_fn = lambda x, y: x function
380 def map_fn(x): function
496 def map_fn(i): function
|
| D | parallel_interleave_test.py | 70 def map_fn(x): function
625 def map_fn(x): function
710 map_fn = lambda x: dataset_ops.Dataset.range(20 * x, 20 * (x + 1)) function
|
| D | from_list_test.py | 68 def testDatasetInputs(self, map_fn): argument
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | flat_map_test.py | 192 def map_fn(x): function
221 def map_fn(y): function
237 def map_fn(x): function
267 def map_fn(x): function
|
| D | reduce_test.py | 121 def map_fn(i): function
204 def map_fn(x): function
|
| D | iterator_test.py | 719 def map_fn(target, handle): function
1057 def map_fn(x): function
|
| D | multi_device_iterator_test.py | 61 def map_fn(v): function
|
| D | shuffle_test.py | 360 def map_fn(x): function
|
| D | batch_test.py | 116 def map_fn(i): function
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_map_ops.py | 27 def map_fn(fn, function
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | batching.py | 446 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
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | traverse_test.py | 92 def map_fn(ds): function
|
| /external/perfetto/include/perfetto/ext/base/threading/ |
| D | stream.h | 68 Stream<FutureReturn<Function, T>> MapFuture(Function map_fn) && { in MapFuture()
|
| D | stream_combinators.h | 78 MapFutureStreamImpl(Stream<T> stream, Function map_fn) in MapFutureStreamImpl()
|
| /external/pytorch/torch/masked/maskedtensor/ |
| D | core.py | 70 def _map_mt_args_kwargs(args, kwargs, map_fn): argument
71 def _helper(a, map_fn): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/saved_model/ |
| D | saved_model.cc | 67 llvm::function_ref<void(const TFRTSavedModelSignatureInfo&)> map_fn) { in MapFunctionSignaturesFromTFSavedModelMLIR()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | map_fn.py | 44 def map_fn(fn, function
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | dataset_creator_model_fit_test_base.py | 55 map_fn = lambda x, y: (lookup_layer(x), y) function
|
| /external/executorch/kernels/portable/cpu/util/ |
| D | kernel_ops_util.h | 142 const MapOp& map_fn, in kernel_reduction_then_map_2d()
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | tf_utils.py | 106 def map_structure_with_atomic(is_atomic_fn, map_fn, nested): argument
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | check_numerics_callback_test.py | 89 def map_fn(x): function
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distributed_table_test.py | 283 def map_fn(vals): function
|
| D | input_lib_test.py | 1159 def map_fn(per_replica_values): function
1296 def map_fn(per_replica_values): function
1421 def map_fn(offset, batch_size): function
|