| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | io.py | 42 def save(dataset, path, compression=None, shard_func=None): argument
88 if shard_func is None:
90 shard_func = lambda *x: None # a dummy function that will not be used function
95 shard_func,
107 shard_func = wrapped_func.function
108 shard_func.add_to_graph(ops.get_default_graph())
115 shard_func_other_args=shard_func.captured_inputs,
117 shard_func=shard_func,
|
| D | snapshot.py | 203 shard_func, argument
227 shard_func,
246 shard_func=self._shard_func.function,
258 def snapshot(path, compression="AUTO", reader_func=None, shard_func=None): argument
338 if shard_func is None:
347 local_shard_func = shard_func
355 shard_func=local_shard_func)
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | SaveDataset.pbtxt | 23 name: "shard_func"
61 name: "shard_func"
|
| D | SnapshotDatasetV2.pbtxt | 47 name: "shard_func"
121 name: "shard_func"
209 name: "shard_func"
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | io_test.py | 66 io.save(dataset, self._test_dir, shard_func=lambda x: x // 21)
76 io.save(dataset, self._test_dir, shard_func=lambda x: x % 7)
|
| D | snapshot_test.py | 169 snapshot.snapshot(self._snapshot_dir, shard_func=lambda _: np.int64(0)))
180 snapshot.snapshot(self._snapshot_dir, shard_func=lambda _: 0))
227 self._snapshot_dir, shard_func=lambda _: "invalid_fn"))
326 snapshot.snapshot(self._snapshot_dir, shard_func=lambda i, _: i % 2))
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SnapshotDatasetV2.pbtxt | 29 name: "shard_func"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.experimental.pbtxt | 193 …argspec: "args=[\'dataset\', \'path\', \'compression\', \'shard_func\'], varargs=None, keywords=No…
205 …argspec: "args=[\'path\', \'compression\', \'reader_func\', \'shard_func\'], varargs=None, keyword…
|
| D | tensorflow.raw_ops.pbtxt | 3889 …argspec: "args=[\'input_dataset\', \'path\', \'shard_func_other_args\', \'shard_func\', \'compress…
4157 …_other_args\', \'output_types\', \'output_shapes\', \'reader_func\', \'shard_func\', \'compression…
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | snapshot_dataset_op.cc | 124 std::unique_ptr<CapturedFunction> shard_func);
313 std::unique_ptr<CapturedFunction> shard_func) in Dataset() argument
322 shard_func_(std::move(shard_func)) { in Dataset()
891 std::unique_ptr<CapturedFunction> shard_func; in MakeDataset() local
894 kShardFuncOtherArgs, &shard_func)); in MakeDataset()
898 std::move(reader_func), std::move(shard_func)); in MakeDataset()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.experimental.pbtxt | 229 …argspec: "args=[\'path\', \'compression\', \'reader_func\', \'shard_func\'], varargs=None, keyword…
|
| D | tensorflow.raw_ops.pbtxt | 3889 …argspec: "args=[\'input_dataset\', \'path\', \'shard_func_other_args\', \'shard_func\', \'compress…
4157 …_other_args\', \'output_types\', \'output_shapes\', \'reader_func\', \'shard_func\', \'compression…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 43775 name: "shard_func"
46656 name: "shard_func"
|