| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| interleave_ops.py | 26 def parallel_interleave(map_func,
34 `parallel_interleave()` maps `map_func` across its input to produce nested
58 map_func: A function mapping a nested structure of tensors to a `Dataset`.
76 map_func, cycle_length, block_length, sloppy, buffer_output_elements,
82 def sloppy_interleave(map_func, cycle_length, block_length=1):
85 `sloppy_interleave()` maps `map_func` across `dataset`, and
118 map_func: A function mapping a nested structure of tensors (having shapes
132 map_func, cycle_length, block_length, sloppy=True)
|
| batching.py | 232 def map_and_batch(map_func,
239 Maps `map_func` across `batch_size` consecutive elements of this dataset
248 map_func: A function mapping a nested structure of tensors to another
272 return batching.map_and_batch(map_func, batch_size, num_parallel_batches,
|
| /external/tensorflow/tensorflow/contrib/eager/python/ |
| saver.py | 48 def restore_variables_on_create(save_path, map_func=None):
63 map_func: A function that given the variable name as argument
74 ValueError: If not used in eager mode or map_func is not callable.
81 if map_func is None:
84 if not callable(map_func):
85 raise ValueError("map_func must be callable.")
86 map_func_wrapper = lambda self, x: map_func(x)
|
| network.py | 577 # The map_func to use (either user-specified or the default).
578 "map_func",
579 # Boolean, True if the user specified an explicit map_func, for error
609 "`map_func=lambda n: n` to save and restore to use fully qualified "
624 ("The map_func passed to restore_network_checkpoint for the Network '%s' "
627 "probably not saved with this map_func. Note that map_func "
666 checkpoint_name = delayed_restoration.map_func(name)
732 """The default map_func for saving or restoring variables.
756 network, save_path, global_step=None, map_func=None)
[all...] |
| saver_test.py | 100 def map_func(x): function in function:SaverTest.testSaveByDict
102 with _saver.restore_variables_on_create(ckpt_prefix, map_func):
|
| network_test.py | 249 map_func=lambda full_name: name_mapping[full_name])
273 load_into, save_path, map_func=_restore_map_func)
350 "The map_func passed to save_network_checkpoint for the Network "
353 make_checkpoint, self.get_temp_dir(), map_func=lambda n: "foo")
357 map_func=lambda n: "foo")
360 loader, checkpoint, map_func=lambda n: "foo")
363 ("The map_func passed to restore_network_checkpoint for the Network"
370 ("The map_func passed to restore_network_checkpoint for the Network"
373 loader, checkpoint, map_func=lambda n: "foo")
[all...] |
| /external/tensorflow/tensorflow/core/kernels/data/ |
| map_dataset_op.cc | 54 MapIteratorFunction map_func; variable
57 map_func = [](IteratorContext* ctx,
66 map_func = [raw_captured_func, indices, can_move](
92 std::move(map_func), preserve_cardinality_);
103 bool use_inter_op_parallelism, MapIteratorFunction map_func,
113 map_func_(std::move(map_func)) {
194 explicit Iterator(const Params& params, MapIteratorFunction map_func)
195 : DatasetIterator<Dataset>(params), map_func_(std::move(map_func)) {}
|
| parallel_map_dataset_op.cc | 251 auto map_func = [this](IteratorContext* ctx, const string& prefix, variable
259 (*ctx->runner())(std::bind(map_func, ctx, prefix,
263 map_func(ctx, prefix, std::move(input_element), result,
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| map_and_filter_fusion.cc | 126 const FunctionDef* map_func =
130 if (!fusion_utils::CanCompose(map_func->signature(),
138 *map_func, *filter_func, "fused_map_and_filter_function",
|
| map_vectorization.cc | 526 const FunctionDef* map_func; local
529 &map_func)) {
534 AddVectorizedFunction(*map_node, *map_func, library);
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| map_and_batch_dataset_serialization_test.py | 50 map_func=_map_fn,
77 map_func=_map_fn,
|
| numa_map_and_batch_dataset_serialization_test.py | 49 map_func=_map_fn,
79 map_func=_map_fn,
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| prefetching_ops.py | 231 def __init__(self, input_dataset, map_func, use_inter_op_parallelism=True):
237 map_func,
260 def map_on_gpu(map_func):
261 """Maps `map_func` across the elements of this dataset.
264 `map_func` on GPU. It must be used after applying the
269 map_func: A function mapping a nested structure of tensors (having shapes
279 return _MapOnGpuDataset(dataset, map_func)
|
| interleave_ops.py | 37 "Use `tf.data.Dataset.interleave(map_func, cycle_length, block_length, "
41 def parallel_interleave(map_func,
49 `parallel_interleave()` maps `map_func` across its input to produce nested
73 map_func: A function mapping a nested structure of tensors to a `Dataset`.
92 dataset, map_func, cycle_length, block_length, sloppy,
|
| batching.py | 568 def __init__(self, input_dataset, map_func, batch_size, num_parallel_calls,
574 map_func,
614 def map_and_batch_with_legacy_function(map_func,
626 map_func: A function mapping a nested structure of tensors to another
661 return _MapAndBatchDataset(dataset, map_func, batch_size,
670 "Use `tf.data.Dataset.map(map_func, num_parallel_calls)` followed by "
674 def map_and_batch(map_func,
681 Maps `map_func` across `batch_size` consecutive elements of this dataset
690 map_func: A function mapping a nested structure of tensors to another
725 return _MapAndBatchDataset(dataset, map_func, batch_size
[all...] |
| /external/tensorflow/tensorflow/python/data/ops/ |
| dataset_ops.py | [all...] |
| readers.py | 125 def __init__(self, input_dataset, map_func, cycle_length, block_length,
130 map_func, self._transformation_name(), dataset=input_dataset)
133 raise TypeError("`map_func` must return a `Dataset` object.")
|
| /external/deqp-deps/SPIRV-Tools/source/opt/ |
| ccp_pass.cpp | 131 auto map_func = [this](uint32_t id) { local
140 map_func);
|
| /external/swiftshader/third_party/SPIRV-Tools/source/opt/ |
| ccp_pass.cpp | 131 auto map_func = [this](uint32_t id) { local
140 map_func);
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| map_and_batch_dataset_op.cc | 92 MapAndBatchIteratorFunction map_func; variable
95 map_func = [](IteratorContext* ctx,
106 map_func = [raw_captured_func, indices, can_move](
140 std::move(map_func), preserve_cardinality_);
153 MapAndBatchIteratorFunction map_func, bool preserve_cardinality)
164 map_func_(std::move(map_func)),
256 MapAndBatchIteratorFunction map_func)
262 map_func_(std::move(map_func)),
[all...] |
| /external/chromium-trace/catapult/devil/devil/utils/ |
| find_usb_devices.py | 390 def CollectHubMaps(hub_types, map_func, device_tree_map=None, fast=False):
395 map_func: [string] Function to run on each hub.
407 yield map_func(hub, hub_type)
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| quantization_mnist_test.py | 193 map_func=_PreprocessFn,
207 map_func=_PreprocessFn,
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| map_and_batch_test.py | 72 map_func=_map_fn,
|
