(*branch_graph_outputs)
| 740 | |
| 741 | |
| 742 | def _get_output_shapes(*branch_graph_outputs): |
| 743 | output_shapes = [] |
| 744 | for out_by_branch in zip(*branch_graph_outputs): |
| 745 | shape = out_by_branch[0].shape |
| 746 | for other_out in out_by_branch[1:]: |
| 747 | shape = shape.most_specific_compatible_shape(other_out.shape) |
| 748 | output_shapes.append(shape) |
| 749 | return output_shapes |
| 750 | |
| 751 | |
| 752 | def verify_captures(op_type, branch_graphs): |
no test coverage detected