Searched refs:branch_operand_shapes (Results 1 – 2 of 2) sorted by relevance
2566 absl::InlinedVector<Shape, 2> branch_operand_shapes; in CreateInstruction() local2567 branch_operand_shapes.reserve(operands.size() - 1); in CreateInstruction()2569 branch_operand_shapes.push_back(operands[i]->shape()); in CreateInstruction()2573 branch_operand_shapes); in CreateInstruction()
2541 std::vector<Shape> branch_operand_shapes(branch_operands.size()); in ConditionalImpl() local2545 TF_ASSIGN_OR_RETURN(branch_operand_shapes[j], in ConditionalImpl()2553 branch_operand_shapes)); in ConditionalImpl()