Searched refs:ShardingSplitProvider (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | split_utils.cc | 60 ShardingSplitProvider::ShardingSplitProvider( in ShardingSplitProvider() function in tensorflow::data::ShardingSplitProvider
68 Status ShardingSplitProvider::GetNext(Tensor* split, bool* end_of_splits) { in GetNext()
82 Status ShardingSplitProvider::Reset() { in Reset()
89 Status ShardingSplitProvider::Save( in Save()
101 Status ShardingSplitProvider::Restore( in Restore()
|
| D | split_utils_test.cc | 94 ShardingSplitProvider split_provider(2, 0, base); in TEST()
101 ShardingSplitProvider split_provider(2, 1, base); in TEST()
108 ShardingSplitProvider split_provider(3, 1, base); in TEST()
115 ShardingSplitProvider split_provider(2, 1, base); in TEST()
124 ShardingSplitProvider split_provider(3, 1, base); in TEST()
|
| D | split_utils.h | 44 class ShardingSplitProvider : public SplitProvider {
46 ShardingSplitProvider(int64 num_shards, int64 shard_index,
|
| D | dataset_test_base.cc | 652 split_provider = absl::make_unique<ShardingSplitProvider>( in CheckSplitProviderShardedIteration()
|