Searched refs:unpruned_branches_ (Results 1 – 2 of 2) sorted by relevance
31 OP_REQUIRES_OK(ctx, ctx->GetAttr("branches", &unpruned_branches_)); in XlaCaseOp()52 return {unpruned_branches_, ctx->Input(0)}; in GetPrunedBranchesAndIndex()56 if (branch_index < 0 || branch_index >= unpruned_branches_.size()) { in GetPrunedBranchesAndIndex()57 branch_index = unpruned_branches_.size() - 1; in GetPrunedBranchesAndIndex()60 std::vector<NameAttrList> pruned_branch = {unpruned_branches_[branch_index]}; in GetPrunedBranchesAndIndex()67 OP_REQUIRES(ctx, !unpruned_branches_.empty(), in Compile()
62 std::vector<NameAttrList> unpruned_branches_; variable