| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | map_vectorization_test.py | 203 map_fn, argument
246 def _testOptimization(self, map_fn, dataset_factory, num_parallel_calls): argument
258 map_fn = lambda x: (x, x + 1) function
268 map_fn = lambda x: x + value function
277 map_fn = lambda x: math_ops.cast(x, dtypes.float64) function
286 map_fn = lambda x: 2 function
295 def map_fn(x): function
308 map_fn = lambda x: array_ops.reshape(x, (-1, 30)) function
334 def testUnaryBitwiseOperations(self, map_fn, num_parallel_calls): argument
343 def testUnaryLogicalOperations(self, map_fn, num_parallel_calls): argument
[all …]
|
| D | hoist_random_uniform_test.py | 89 def testHoistFunction(self, map_fn, should_optimize): argument
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | preprocessing_stage.py | 76 def map_fn(x): function
210 def map_fn(*x): function
214 def map_fn(*x): function
270 def map_fn(*x, j=i): function
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | flat_map_dataset_serialization_test.py | 47 def map_fn(x): function
61 def map_fn(y): function
75 def map_fn(x): function
105 def map_fn(x): function
|
| D | choose_fastest_dataset_serialization_test.py | 41 map_fn = lambda x: x * 2 function
|
| D | map_and_batch_dataset_serialization_test.py | 95 def map_fn(i): 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
|
| D | parallel_interleave_test.py | 72 def map_fn(x): function
626 def map_fn(x): function
711 map_fn = lambda x: dataset_ops.Dataset.range(20 * x, 20 * (x + 1)) function
|
| D | optimize_dataset_test.py | 59 def map_fn(x): function
|
| D | snapshot_test.py | 565 def map_fn(x): function
|
| /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
85 const string& map_fn, int num_parallel_calls = 0) { in AddMapNode()
294 auto map_fn = AddMapFn(&graph); in TEST_P() local
342 const string& input_dataset, const string& map_fn, in AddMapAndBatchNode()
371 auto map_fn = AddMapFn(&graph); in TEST_P() local
404 auto map_fn = AddMapFn(&graph); in TEST_P() local
521 auto map_fn = AddMapFn(&graph); in TEST() local
540 auto map_fn = AddMapFn(&graph); in TEST() local
559 auto map_fn = AddMapFn(&graph); in TEST() local
577 auto map_fn = AddMapFn(&graph); in TEST() local
|
| /external/tensorflow/tensorflow/python/data/experimental/benchmarks/ |
| D | map_vectorization_benchmark.py | 108 def _compare(self, input_dataset, map_fn, batch_size, input_size, str_id): argument
176 def _benchmark_helper(self, map_fn, str_id, base_dataset_factory=None): argument
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_map_ops.py | 30 def map_fn(fn, function
|
| /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
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | traverse_test.py | 91 def map_fn(ds): function
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | reduce_test.py | 125 def map_fn(i): function
208 def map_fn(x): function
|
| D | batch_test.py | 111 def map_fn(i): function
|
| D | iterator_test.py | 693 def map_fn(target, handle): function
1004 def map_fn(x): function
|
| D | shuffle_test.py | 316 def map_fn(x): function
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_integration_test.py | 154 def map_fn(_): function
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | map_fn.py | 47 def map_fn(fn, function
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | tf_utils.py | 107 def map_structure_with_atomic(is_atomic_fn, map_fn, nested): argument
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | check_numerics_callback_test.py | 93 def map_fn(x): function
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib_test.py | 1095 def map_fn(per_replica_values): function
1238 def map_fn(per_replica_values): function
1359 def map_fn(offset, batch_size): function
|
| /external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/ |
| D | rnn_cell_wrapper_impl.py | 507 def _enumerated_map_structure_up_to(shallow_structure, map_fn, *args, **kwargs): argument
|