Home
last modified time | relevance | path

Searched defs:map_fn (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dpreprocessing_stage.py73 def map_fn(x): function
206 def map_fn(*x): function
210 def map_fn(*x): function
266 def map_fn(*x, j=i): function
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dmap_and_batch_test.py334 map_fn = lambda x: x function
343 map_fn = lambda x: (x, x) function
352 map_fn = lambda x, y: (y, x) function
363 map_fn = lambda x, y: x function
384 def map_fn(x): function
500 def map_fn(i): function
Dparallel_interleave_test.py74 def map_fn(x): function
628 def map_fn(x): function
713 map_fn = lambda x: dataset_ops.Dataset.range(20 * x, 20 * (x + 1)) function
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dflat_map_test.py187 def map_fn(x): function
216 def map_fn(y): function
232 def map_fn(x): function
262 def map_fn(x): function
Dreduce_test.py125 def map_fn(i): function
208 def map_fn(x): function
Diterator_test.py693 def map_fn(target, handle): function
1004 def map_fn(x): function
Dmulti_device_iterator_test.py65 def map_fn(v): function
Dbatch_test.py119 def map_fn(i): function
Dshuffle_test.py364 def map_fn(x): function
Ddataset_test.py619 def map_fn(x): function
Dsnapshot_test.py578 def map_fn(x): function
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_map_ops.py30 def map_fn(fn, function
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dbatching.py441 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/
Dtraverse_test.py95 def map_fn(ds): function
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_integration_test.py150 def map_fn(_): function
/external/tensorflow/tensorflow/compiler/mlir/tfrt/saved_model/
Dsaved_model.cc67 llvm::function_ref<void(const TFRTSavedModelSignatureInfo&)> map_fn) { in MapFunctionSignaturesFromTFSavedModelMLIR()
/external/tensorflow/tensorflow/python/ops/
Dmap_fn.py47 def map_fn(fn, function
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Doptimization_test.py57 def map_fn(x): function
/external/tensorflow/tensorflow/python/keras/distribute/
Ddataset_creator_model_fit_test_base.py55 map_fn = lambda x, y: (lookup_layer(x), y) function
/external/tensorflow/tensorflow/python/keras/utils/
Dtf_utils.py106 def map_structure_with_atomic(is_atomic_fn, map_fn, nested): argument
/external/tensorflow/tensorflow/python/distribute/
Dinput_lib_test.py1119 def map_fn(per_replica_values): function
1261 def map_fn(per_replica_values): function
1382 def map_fn(offset, batch_size): function
/external/tensorflow/tensorflow/python/debug/lib/
Dcheck_numerics_callback_test.py93 def map_fn(x): function
/external/tensorflow/tensorflow/python/distribute/coordinator/
Dcluster_coordinator_test.py579 def map_fn(x): function
1366 def map_fn(x): function
/external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/
Drnn_cell_wrapper_impl.py507 def _enumerated_map_structure_up_to(shallow_structure, map_fn, *args, **kwargs): argument
/external/tensorflow/tensorflow/python/framework/
Dop_callbacks_test.py664 def map_fn(x): function

12