Searched defs:shard_func (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | save_op.py | 39 shard_func=None, argument
75 def __init__(self, dataset, path, shard_func, compression): argument
100 def set_save_dataset_attributes(dataset, shard_func, path): argument
104 shard_func = lambda *x: None # a dummy function that will not be used function
|
| D | dataset_ops.py | 1690 shard_func=None, argument
3142 shard_func=None, argument
6204 shard_func, argument
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | io.py | 37 shard_func=None, argument
|
| D | snapshot.py | 192 def snapshot(path, compression="AUTO", reader_func=None, shard_func=None): argument
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | io_ops_test.cc | 39 FunctionDefHelper::AttrValueWrapper shard_func, in SaveDatasetV2Params()
|
| D | io_ops.cc | 218 std::unique_ptr<CapturedFunction> shard_func, bool use_shard_func) in Dataset()
545 std::unique_ptr<CapturedFunction> shard_func; in MakeDataset() local
|
| D | snapshot_dataset_op.cc | 124 std::unique_ptr<CapturedFunction> shard_func) in Dataset()
788 std::unique_ptr<CapturedFunction> shard_func; in MakeDataset() local
|