Home
last modified time | relevance | path

Searched refs:map_func (Results 1 – 25 of 39) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/eager/python/
Dnetwork.py666 checkpoint_name = delayed_restoration.map_func(name)
756 network, save_path, global_step=None, map_func=None): argument
792 user_map_func = map_func
793 if map_func is None:
794 map_func = _make_prefix_stripping_map_fn(network.scope_name)
797 mapped_name = map_func(variable._shared_name)
866 def _restore_existing_variables(network, save_path, map_func, user_map_func): argument
885 checkpoint_name = map_func(variable._shared_name)
914 def _set_restore_on_create(network, save_path, map_func, user_map_func, argument
942 map_func=map_func,
[all …]
Dsaver.py48 def restore_variables_on_create(save_path, map_func=None): argument
81 if map_func is None:
84 if not callable(map_func):
86 map_func_wrapper = lambda self, x: map_func(x)
/external/tensorflow/tensorflow/contrib/data/python/ops/
Dinterleave_ops.py26 def parallel_interleave(map_func, argument
76 map_func, cycle_length, block_length, sloppy, buffer_output_elements,
82 def sloppy_interleave(map_func, cycle_length, block_length=1): argument
132 map_func, cycle_length, block_length, sloppy=True)
Dbatching.py232 def map_and_batch(map_func, argument
272 return batching.map_and_batch(map_func, batch_size, num_parallel_batches,
/external/tensorflow/tensorflow/core/kernels/data/
Dmap_dataset_op.cc54 MapIteratorFunction map_func; in MakeDataset() local
57 map_func = [](IteratorContext* ctx, in MakeDataset()
66 map_func = [raw_captured_func, indices, can_move]( in MakeDataset()
92 std::move(map_func), preserve_cardinality_); in MakeDataset()
103 bool use_inter_op_parallelism, MapIteratorFunction map_func, in Dataset() argument
113 map_func_(std::move(map_func)) { in Dataset()
194 explicit Iterator(const Params& params, MapIteratorFunction map_func) in Iterator() argument
195 : DatasetIterator<Dataset>(params), map_func_(std::move(map_func)) {} in Iterator()
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py929 def map(self, map_func, num_parallel_calls=None): argument
1022 return MapDataset(self, map_func, preserve_cardinality=True)
1025 self, map_func, num_parallel_calls, preserve_cardinality=True)
1027 def flat_map(self, map_func): argument
1055 return FlatMapDataset(self, map_func)
1058 map_func, argument
1135 return InterleaveDataset(self, map_func, cycle_length, block_length)
1137 return ParallelInterleaveDataset(self, map_func, cycle_length,
1661 def map(self, map_func, num_parallel_calls=None): argument
1664 MapDataset(self, map_func, preserve_cardinality=False))
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.-dataset.pbtxt48 argspec: "args=[\'self\', \'map_func\'], varargs=None, keywords=None, defaults=None"
68 …argspec: "args=[\'self\', \'map_func\', \'cycle_length\', \'block_length\', \'num_parallel_calls\'…
84 …argspec: "args=[\'self\', \'map_func\', \'num_parallel_calls\'], varargs=None, keywords=None, defa…
88 …argspec: "args=[\'self\', \'map_func\', \'num_parallel_calls\'], varargs=None, keywords=None, defa…
Dtensorflow.data.-t-f-record-dataset.pbtxt50 argspec: "args=[\'self\', \'map_func\'], varargs=None, keywords=None, defaults=None"
70 …argspec: "args=[\'self\', \'map_func\', \'cycle_length\', \'block_length\', \'num_parallel_calls\'…
86 …argspec: "args=[\'self\', \'map_func\', \'num_parallel_calls\'], varargs=None, keywords=None, defa…
90 …argspec: "args=[\'self\', \'map_func\', \'num_parallel_calls\'], varargs=None, keywords=None, defa…
Dtensorflow.data.experimental.-csv-dataset.pbtxt50 argspec: "args=[\'self\', \'map_func\'], varargs=None, keywords=None, defaults=None"
70 …argspec: "args=[\'self\', \'map_func\', \'cycle_length\', \'block_length\', \'num_parallel_calls\'…
86 …argspec: "args=[\'self\', \'map_func\', \'num_parallel_calls\'], varargs=None, keywords=None, defa…
90 …argspec: "args=[\'self\', \'map_func\', \'num_parallel_calls\'], varargs=None, keywords=None, defa…
Dtensorflow.data.-text-line-dataset.pbtxt50 argspec: "args=[\'self\', \'map_func\'], varargs=None, keywords=None, defaults=None"
70 …argspec: "args=[\'self\', \'map_func\', \'cycle_length\', \'block_length\', \'num_parallel_calls\'…
86 …argspec: "args=[\'self\', \'map_func\', \'num_parallel_calls\'], varargs=None, keywords=None, defa…
90 …argspec: "args=[\'self\', \'map_func\', \'num_parallel_calls\'], varargs=None, keywords=None, defa…
Dtensorflow.data.experimental.-sql-dataset.pbtxt50 argspec: "args=[\'self\', \'map_func\'], varargs=None, keywords=None, defaults=None"
70 …argspec: "args=[\'self\', \'map_func\', \'cycle_length\', \'block_length\', \'num_parallel_calls\'…
86 …argspec: "args=[\'self\', \'map_func\', \'num_parallel_calls\'], varargs=None, keywords=None, defa…
90 …argspec: "args=[\'self\', \'map_func\', \'num_parallel_calls\'], varargs=None, keywords=None, defa…
Dtensorflow.data.-fixed-length-record-dataset.pbtxt50 argspec: "args=[\'self\', \'map_func\'], varargs=None, keywords=None, defaults=None"
70 …argspec: "args=[\'self\', \'map_func\', \'cycle_length\', \'block_length\', \'num_parallel_calls\'…
86 …argspec: "args=[\'self\', \'map_func\', \'num_parallel_calls\'], varargs=None, keywords=None, defa…
90 …argspec: "args=[\'self\', \'map_func\', \'num_parallel_calls\'], varargs=None, keywords=None, defa…
Dtensorflow.data.experimental.-random-dataset.pbtxt50 argspec: "args=[\'self\', \'map_func\'], varargs=None, keywords=None, defaults=None"
70 …argspec: "args=[\'self\', \'map_func\', \'cycle_length\', \'block_length\', \'num_parallel_calls\'…
86 …argspec: "args=[\'self\', \'map_func\', \'num_parallel_calls\'], varargs=None, keywords=None, defa…
90 …argspec: "args=[\'self\', \'map_func\', \'num_parallel_calls\'], varargs=None, keywords=None, defa…
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dprefetching_ops.py231 def __init__(self, input_dataset, map_func, use_inter_op_parallelism=True): argument
237 map_func,
260 def map_on_gpu(map_func): argument
279 return _MapOnGpuDataset(dataset, map_func)
Dbatching.py568 def __init__(self, input_dataset, map_func, batch_size, num_parallel_calls, argument
574 map_func,
614 def map_and_batch_with_legacy_function(map_func, argument
661 return _MapAndBatchDataset(dataset, map_func, batch_size,
674 def map_and_batch(map_func, argument
725 return _MapAndBatchDataset(dataset, map_func, batch_size,
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_and_filter_fusion.cc126 const FunctionDef* map_func = in OptimizeAndCollectStats() local
130 if (!fusion_utils::CanCompose(map_func->signature(), in OptimizeAndCollectStats()
138 *map_func, *filter_func, "fused_map_and_filter_function", in OptimizeAndCollectStats()
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.data.-dataset.pbtxt31 argspec: "args=[\'self\', \'map_func\'], varargs=None, keywords=None, defaults=None"
47 …argspec: "args=[\'self\', \'map_func\', \'cycle_length\', \'block_length\', \'num_parallel_calls\'…
55 …argspec: "args=[\'self\', \'map_func\', \'num_parallel_calls\'], varargs=None, keywords=None, defa…
Dtensorflow.data.experimental.-random-dataset.pbtxt33 argspec: "args=[\'self\', \'map_func\'], varargs=None, keywords=None, defaults=None"
49 …argspec: "args=[\'self\', \'map_func\', \'cycle_length\', \'block_length\', \'num_parallel_calls\'…
57 …argspec: "args=[\'self\', \'map_func\', \'num_parallel_calls\'], varargs=None, keywords=None, defa…
Dtensorflow.data.-text-line-dataset.pbtxt33 argspec: "args=[\'self\', \'map_func\'], varargs=None, keywords=None, defaults=None"
49 …argspec: "args=[\'self\', \'map_func\', \'cycle_length\', \'block_length\', \'num_parallel_calls\'…
57 …argspec: "args=[\'self\', \'map_func\', \'num_parallel_calls\'], varargs=None, keywords=None, defa…
Dtensorflow.data.-t-f-record-dataset.pbtxt32 argspec: "args=[\'self\', \'map_func\'], varargs=None, keywords=None, defaults=None"
48 …argspec: "args=[\'self\', \'map_func\', \'cycle_length\', \'block_length\', \'num_parallel_calls\'…
56 …argspec: "args=[\'self\', \'map_func\', \'num_parallel_calls\'], varargs=None, keywords=None, defa…
Dtensorflow.data.experimental.-sql-dataset.pbtxt33 argspec: "args=[\'self\', \'map_func\'], varargs=None, keywords=None, defaults=None"
49 …argspec: "args=[\'self\', \'map_func\', \'cycle_length\', \'block_length\', \'num_parallel_calls\'…
57 …argspec: "args=[\'self\', \'map_func\', \'num_parallel_calls\'], varargs=None, keywords=None, defa…
Dtensorflow.data.-fixed-length-record-dataset.pbtxt33 argspec: "args=[\'self\', \'map_func\'], varargs=None, keywords=None, defaults=None"
49 …argspec: "args=[\'self\', \'map_func\', \'cycle_length\', \'block_length\', \'num_parallel_calls\'…
57 …argspec: "args=[\'self\', \'map_func\', \'num_parallel_calls\'], varargs=None, keywords=None, defa…
Dtensorflow.data.experimental.-csv-dataset.pbtxt33 argspec: "args=[\'self\', \'map_func\'], varargs=None, keywords=None, defaults=None"
49 …argspec: "args=[\'self\', \'map_func\', \'cycle_length\', \'block_length\', \'num_parallel_calls\'…
57 …argspec: "args=[\'self\', \'map_func\', \'num_parallel_calls\'], varargs=None, keywords=None, defa…
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dmap_and_batch_dataset_serialization_test.py50 map_func=_map_fn,
77 map_func=_map_fn,
Dnuma_map_and_batch_dataset_serialization_test.py49 map_func=_map_fn,
79 map_func=_map_fn,

12