Searched refs:branch_index_ (Results 1 – 2 of 2) sorted by relevance
365 if (branch_index_ < dataset()->captured_funcs_.size()) { in GetNextInternal()368 TF_RETURN_IF_ERROR(MakeCurrentIterator(ctx, branch_index_, in GetNextInternal()381 branch_index_++; in GetNextInternal()416 writer->WriteScalar(full_name("branch_index"), branch_index_)); in SaveInternal()435 reader->ReadScalar(full_name("branch_index"), &branch_index_)); in RestoreInternal()441 if (branch_index_ < dataset()->captured_funcs_.size()) { in RestoreInternal()442 TF_RETURN_IF_ERROR(MakeCurrentIterator(ctx, branch_index_, in RestoreInternal()460 DCHECK_GE(branch_index_, 0); in GetNextFromExperiment()461 DCHECK_LT(branch_index_, histograms_.size()); in GetNextFromExperiment()470 histograms_[branch_index_].Add( in GetNextFromExperiment()[all …]
79 OutputTensor branch_index_; member in tensorflow::__anon8ede7fbe0111::CaseBuilder115 TF_CHECK_OK(case_op_->input_tensor(0, &branch_index_)); in CaseBuilder()124 .Input(NodeOut(branch_index_)) in CreatePivotNodes()125 .Input(NodeOut(branch_index_)) in CreatePivotNodes()161 .Input(branch_index_) in AddInput()