Home
last modified time | relevance | path

Searched refs:reader_func (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/ops/
Dio.py125 reader_func=None): argument
127 if reader_func is None:
128 reader_func = lambda datasets: datasets.interleave( # pylint:disable=g-long-lambda function
146 reader_func,
156 reader_func=self._reader_func.function,
169 def load(path, element_spec=None, compression=None, reader_func=None): argument
232 reader_func=reader_func)
Dsnapshot.py205 reader_func=None, argument
209 if reader_func is None:
210 reader_func = lambda datasets: datasets.interleave( # pylint:disable=g-long-lambda function
220 reader_func,
245 reader_func=self._reader_func.function,
258 def snapshot(path, compression="AUTO", reader_func=None, shard_func=None): argument
352 reader_func=reader_func,
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DSnapshotDatasetV2.pbtxt43 name: "reader_func"
117 name: "reader_func"
205 name: "reader_func"
DLoadDataset.pbtxt35 name: "reader_func"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SnapshotDatasetV2.pbtxt23 name: "reader_func"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.data.experimental.pbtxt153 …argspec: "args=[\'path\', \'element_spec\', \'compression\', \'reader_func\'], varargs=None, keywo…
205 …argspec: "args=[\'path\', \'compression\', \'reader_func\', \'shard_func\'], varargs=None, keyword…
Dtensorflow.raw_ops.pbtxt2137 …\', \'reader_func_other_args\', \'output_types\', \'output_shapes\', \'reader_func\', \'compressio…
4157 …s\', \'shard_func_other_args\', \'output_types\', \'output_shapes\', \'reader_func\', \'shard_func…
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dio_test.py80 reader_func=lambda x: x.flat_map(lambda y: y))
Dsnapshot_test.py197 reader_func=(
214 reader_func=(
236 snapshot.snapshot(self._snapshot_dir, reader_func=lambda x: x + 1))
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dsnapshot_dataset_op.cc123 std::unique_ptr<CapturedFunction> reader_func,
312 std::unique_ptr<CapturedFunction> reader_func, in Dataset() argument
321 reader_func_(std::move(reader_func)), in Dataset()
887 std::unique_ptr<CapturedFunction> reader_func; in MakeDataset() local
890 kReaderFuncOtherArgs, &reader_func)); in MakeDataset()
898 std::move(reader_func), std::move(shard_func)); in MakeDataset()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.experimental.pbtxt229 …argspec: "args=[\'path\', \'compression\', \'reader_func\', \'shard_func\'], varargs=None, keyword…
Dtensorflow.raw_ops.pbtxt2137 …\', \'reader_func_other_args\', \'output_types\', \'output_shapes\', \'reader_func\', \'compressio…
4157 …s\', \'shard_func_other_args\', \'output_types\', \'output_shapes\', \'reader_func\', \'shard_func…
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt21401 name: "reader_func"
46652 name: "reader_func"