(shape_list, enqueue_many)
| 668 | |
| 669 | |
| 670 | def _merge_shapes(shape_list, enqueue_many): |
| 671 | shape_list = [tensor_shape.as_shape(s) for s in shape_list] |
| 672 | if enqueue_many: |
| 673 | # We want the shapes without the leading batch dimension. |
| 674 | shape_list = [s.with_rank_at_least(1)[1:] for s in shape_list] |
| 675 | merged_shape = shape_list[0] |
| 676 | for s in shape_list[1:]: |
| 677 | merged_shape.merge_with(s) |
| 678 | return merged_shape.as_list() |
| 679 | |
| 680 | |
| 681 | def _shapes(tensor_list_list, shapes, enqueue_many): |
no test coverage detected