Searched refs:ShardingSplitProvider (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/data/ |
| D | split_utils_test.cc | 98 ShardingSplitProvider split_provider(2, 0, base); in TEST()
105 ShardingSplitProvider split_provider(2, 1, base); in TEST()
112 ShardingSplitProvider split_provider(3, 1, base); in TEST()
119 ShardingSplitProvider split_provider(2, 1, base); in TEST()
128 ShardingSplitProvider split_provider(3, 1, base); in TEST()
|
| D | split_utils.cc | 66 ShardingSplitProvider::ShardingSplitProvider( in ShardingSplitProvider() function in tensorflow::data::ShardingSplitProvider
74 Status ShardingSplitProvider::GetNext(Tensor* split, bool* end_of_splits) { in GetNext()
88 Status ShardingSplitProvider::Reset() { in Reset()
95 Status ShardingSplitProvider::Save( in Save()
107 Status ShardingSplitProvider::Restore( in Restore()
|
| D | split_utils.h | 47 class ShardingSplitProvider : public SplitProvider {
49 ShardingSplitProvider(int64_t num_shards, int64_t shard_index,
|
| D | dataset_test_base.cc | 658 split_providers[i] = std::make_unique<ShardingSplitProvider>( in CheckSplitProviderShardedIteration()
|