Searched refs:IndexSplitProvider (Results 1 – 4 of 4) sorted by relevance
56 IndexSplitProvider split_provider(0); in TEST()62 IndexSplitProvider split_provider(1); in TEST()68 IndexSplitProvider split_provider(3); in TEST()74 IndexSplitProvider split_provider(4); in TEST()93 auto base = std::make_shared<IndexSplitProvider>(4); in TEST()100 auto base = std::make_shared<IndexSplitProvider>(4); in TEST()107 auto base = std::make_shared<IndexSplitProvider>(6); in TEST()114 auto base = std::make_shared<IndexSplitProvider>(1); in TEST()121 auto base = std::make_shared<IndexSplitProvider>(6); in TEST()
26 IndexSplitProvider::IndexSplitProvider(int64 n) : i_(0), n_(n) {} in IndexSplitProvider() function in tensorflow::data::IndexSplitProvider28 Status IndexSplitProvider::GetNext(Tensor* split, bool* end_of_splits) { in GetNext()40 Status IndexSplitProvider::Reset() { in Reset()46 Status IndexSplitProvider::Save( in Save()53 Status IndexSplitProvider::Restore( in Restore()
26 class IndexSplitProvider : public SplitProvider {28 explicit IndexSplitProvider(int64 n);
64 absl::make_unique<IndexSplitProvider>(tensors_[0].dim_size(0)); in MakeSplitProvider()119 split_provider_ = std::make_shared<IndexSplitProvider>( in Initialize()