(dim)
| 777 | if rcond is None: |
| 778 | |
| 779 | def get_dim_size(dim): |
| 780 | dim_val = tensor_shape.dimension_value(a.shape[dim]) |
| 781 | if dim_val is not None: |
| 782 | return dim_val |
| 783 | return array_ops.shape(a)[dim] |
| 784 | |
| 785 | num_rows = get_dim_size(-2) |
| 786 | num_cols = get_dim_size(-1) |