Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dmap_vectorization_test.py326 map_fn, argument
371 def testOptimization(self, map_fn, base_dataset_factory, num_parallel_calls): argument
379 def map_fn(x): function
396 def map_fn(x): function
410 def map_fn(x): function
439 def map_fn(x): function
485 def map_fn(x): function
503 def map_fn(x): function
514 def map_fn(x): function
547 def map_fn(x): function
Doptimize_dataset_test.py62 def map_fn(x): function
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dflat_map_dataset_serialization_test.py41 def map_fn(x): function
54 def map_fn(y): function
67 def map_fn(x): function
95 def map_fn(x): function
Dchoose_fastest_dataset_serialization_test.py35 map_fn = lambda x: x * 2 function
Dmap_and_batch_dataset_serialization_test.py91 def map_fn(i): function
/external/tensorflow/tensorflow/python/data/benchmarks/
Dmap_benchmark.py30 def benchmark_helper(chain_length, map_fn, use_inter_op_parallelism, label): argument
49 def benchmark_helper(fan_out, map_fn, use_inter_op_parallelism, label): argument
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_vectorization_test.cc59 FunctionDef* map_fn = graph->graph()->mutable_library()->add_function(); in AddMapFn() local
72 const string& map_fn, int num_parallel_calls = 0) { in AddMapNode()
246 auto map_fn = AddMapFn(&graph); in TEST_P() local
285 const string& input_dataset, const string& map_fn, in AddMapAndBatchNode()
314 auto map_fn = AddMapFn(&graph); in TEST() local
340 auto map_fn = AddMapFn(&graph); in TEST_P() local
433 auto map_fn = AddMapFn(&graph); in TEST() local
453 auto map_fn = AddMapFn(&graph); in TEST() local
473 auto map_fn = AddMapFn(&graph); in TEST() local
492 auto map_fn = AddMapFn(&graph); in TEST() local
Dmap_vectorization.cc453 const FunctionDef*& map_fn = *map_fn_output; in FindMapAndBatchPattern() local
/external/tensorflow/tensorflow/python/data/experimental/benchmarks/
Dmap_vectorization_benchmark.py115 def _compare(self, input_dataset, map_fn, batch_size, input_size, str_id): argument
185 def _benchmark_helper(self, map_fn, str_id, base_dataset_factory=None): argument
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dmap_and_batch_test.py438 def testShortCircuit(self, structure, map_fn, num_parallel_calls): argument
465 def map_fn(x): function
Dparallel_interleave_test.py72 def map_fn(x): function
641 def map_fn(x): function
724 map_fn = lambda x: dataset_ops.Dataset.range(20 * x, 20 * (x + 1)) function
/external/tensorflow/tensorflow/python/data/util/
Dtraverse_test.py91 def map_fn(ds): function
/external/tensorflow/tensorflow/python/ops/
Dmap_fn.py38 def map_fn(fn, elems, dtype=None, parallel_iterations=None, back_prop=True, function
Drnn_cell_impl.py1072 def _enumerated_map_structure_up_to(shallow_structure, map_fn, *args, **kwargs): argument
Dcontrol_flow_ops.py2933 def map_fn(x): function
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_map_ops.py37 def map_fn(fn, function
/external/tensorflow/tensorflow/python/keras/utils/
Dtf_utils.py145 def map_structure_with_atomic(is_atomic_fn, map_fn, nested): argument
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dreduce_test.py117 def map_fn(i): function
Diterator_test.py690 def map_fn(target, handle): function
Dmap_test.py77 def map_fn(x): function
933 def testShortCircuit(self, structure, map_fn, num_parallel_calls): argument
/external/v8/tools/
Dmap-processor.js676 function BreakDown(list, map_fn) { argument
678 map_fn = each => each; function in BreakDown
/external/tensorflow/tensorflow/contrib/labeled_tensor/
D__init__.py110 map_fn = _ops.map_fn variable
/external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/
Dops.py611 def map_fn(fn, labeled_tensor, name=None): function
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dreaders.py480 def map_fn(*columns): function
/external/v8/src/builtins/
Dbuiltins-typed-array-gen.cc1646 TNode<Object> map_fn = args.GetOptionalArgumentValue(1); in TF_BUILTIN() local

12