Searched refs:take_count_ (Results 1 – 4 of 4) sorted by relevance
32 TakeNode::TakeNode(std::shared_ptr<DatasetNode> child, int32_t count) : take_count_(count) { this->… in TakeNode()35 auto node = std::make_shared<TakeNode>(nullptr, take_count_); in Copy()39 …nt(std::ostream &out) const { out << (Name() + "(num_rows:" + std::to_string(take_count_) + ")"); } in Print()43 auto op = std::make_shared<TakeOp>(take_count_); in Build()53 if (take_count_ <= 0 && take_count_ != -1) { in ValidateParams()55 …e: take_count should be either -1 or positive integer, take_count: " + std::to_string(take_count_); in ValidateParams()71 …*dataset_size = take_count_ == -1 ? num_rows : std::min(static_cast<int64_t>(take_count_), num_row… in GetDatasetSize()90 args["count"] = take_count_; in to_json()
60 const int32_t Count() const { return take_count_; } in Count()84 int32_t TakeCount() const { return take_count_; } in TakeCount()100 int32_t take_count_;
25 TakeOp::TakeOp(int32_t count) : PipelineOp(0), max_takes_(count), take_count_(0) {} in TakeOp()38 out << "\nTake count: " << take_count_ << "\nMax takes: " << max_takes_ << "\n\n"; in Print()46 if (take_count_ < max_takes_) { in GetNextRow()51 take_count_++; in GetNextRow()55 if (take_count_ == max_takes_) { in GetNextRow()64 take_count_ = 0; in GetNextRow()
67 int32_t take_count_; // A counter for the current number of executed takes variable