Home
last modified time | relevance | path

Searched defs:shard_func (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/ops/
Dio.py42 def save(dataset, path, compression=None, shard_func=None): argument
90 shard_func = lambda *x: None # a dummy function that will not be used function
Dsnapshot.py203 shard_func, argument
258 def snapshot(path, compression="AUTO", reader_func=None, shard_func=None): argument
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dsnapshot_dataset_op.cc313 std::unique_ptr<CapturedFunction> shard_func) in Dataset()
891 std::unique_ptr<CapturedFunction> shard_func; in MakeDataset() local