Searched refs:branch_computation_shapes (Results 1 – 2 of 2) sorted by relevance
2560 absl::InlinedVector<ProgramShape, 2> branch_computation_shapes; in CreateInstruction() local2561 branch_computation_shapes.reserve(branch_computations->size()); in CreateInstruction()2563 branch_computation_shapes.push_back( in CreateInstruction()2572 operands[0]->shape(), branch_computation_shapes, in CreateInstruction()
2542 std::vector<ProgramShape> branch_computation_shapes( in ConditionalImpl() local2547 TF_ASSIGN_OR_RETURN(branch_computation_shapes[j], in ConditionalImpl()2552 *branch_index_shape, branch_computation_shapes, in ConditionalImpl()