| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| H A 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
|
| H A D | optimization_test.py | 52 def map_fn(x): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| H A 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
|
| H A 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
|
| H A D | from_list_test.py | 68 def testDatasetInputs(self, map_fn): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/ |
| H A 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
|
| H A D | reduce_test.py | 121 def map_fn(i): function
204 def map_fn(x): function
|
| H A D | iterator_test.py | 719 def map_fn(target, handle): function
1057 def map_fn(x): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/ |
| H A 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
|
| /aosp_15_r20/external/pytorch/torch/masked/maskedtensor/ |
| H A D | core.py | 70 def _map_mt_args_kwargs(args, kwargs, map_fn): argument
71 def _helper(a, map_fn): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_map_ops.py | 27 def map_fn(fn, function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/util/ |
| H A D | traverse_test.py | 92 def map_fn(ds): function
|
| /aosp_15_r20/prebuilts/runtime/mainline/platform/sdk/include/external/perfetto/include/perfetto/ext/base/threading/ |
| H A D | stream.h | 68 Stream<FutureReturn<Function, T>> MapFuture(Function map_fn) && { in MapFuture()
|
| H A D | stream_combinators.h | 78 MapFutureStreamImpl(Stream<T> stream, Function map_fn) in MapFutureStreamImpl()
|
| /aosp_15_r20/external/perfetto/include/perfetto/ext/base/threading/ |
| H A D | stream.h | 68 Stream<FutureReturn<Function, T>> MapFuture(Function map_fn) && { in MapFuture()
|
| H A D | stream_combinators.h | 78 MapFutureStreamImpl(Stream<T> stream, Function map_fn) in MapFutureStreamImpl()
|
| /aosp_15_r20/external/executorch/exir/emit/test/ |
| H A D | test_emit.py | 639 def map_fn(x: torch.Tensor, y: torch.Tensor) -> torch.Tensor: function
730 def map_fn(x: torch.Tensor, y: torch.Tensor) -> torch.Tensor: function
747 def map_fn(x: torch.Tensor, y: torch.Tensor) -> torch.Tensor: function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/saved_model/ |
| H A D | saved_model.cc | 67 llvm::function_ref<void(const TFRTSavedModelSignatureInfo&)> map_fn) { in MapFunctionSignaturesFromTFSavedModelMLIR()
|
| /aosp_15_r20/external/executorch/exir/backend/test/ |
| H A D | test_backends.py | 1080 def map_fn(x, y): function
1146 def map_fn(x, pred1, pred2, y): function
|
| H A D | test_backends_lifted.py | 1067 def map_fn(x, y): function
1134 def map_fn(x, pred1, pred2, y): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | map_fn.py | 44 def map_fn(fn, function
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_control_flow.py | 2977 def map_fn(x, y): function
3003 def map_fn(x, y): function
3035 def map_fn(x, y): function
3052 def map_fn(x, y): function
3092 def map_fn(x): function
3424 def map_fn(pred, x): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/coordinator/ |
| H A D | cluster_coordinator_test.py | 615 def map_fn(x): function
1438 def map_fn(x): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/distribute/ |
| H A D | dataset_creator_model_fit_test_base.py | 55 map_fn = lambda x, y: (lookup_layer(x), y) function
|