Helper to create a WrappedTensor object.
(tensor, is_stacked=True, is_sparse_stacked=False)
| 931 | |
| 932 | |
| 933 | def wrap(tensor, is_stacked=True, is_sparse_stacked=False): |
| 934 | """Helper to create a WrappedTensor object.""" |
| 935 | assert isinstance(is_stacked, bool) |
| 936 | assert isinstance(is_sparse_stacked, bool) |
| 937 | assert isinstance(tensor, ops.Tensor) |
| 938 | assert not is_sparse_stacked or is_stacked, ("If the wrapped tensor is " |
| 939 | "stacked via a sparse " |
| 940 | "conversion, it must also be " |
| 941 | "stacked.") |
| 942 | return WrappedTensor(tensor, is_stacked, is_sparse_stacked) |
| 943 | |
| 944 | |
| 945 | def _fallback_converter(pfor_input): |
no test coverage detected