| 35 | case 1: |
| 36 | return core::TensorShape::from_dims({dims[0]}); |
| 37 | case 2: |
| 38 | return core::TensorShape::from_dims({dims[0], dims[1]}); |
| 39 | case 3: |
| 40 | return core::TensorShape::from_dims({dims[0], dims[1], dims[2]}); |
| 41 | case 4: |
| 42 | return core::TensorShape::from_dims({dims[0], dims[1], dims[2], dims[3]}); |
| 43 | default: |
| 44 | throw std::runtime_error("unsupported tensor rank"); |
| 45 | } |
| 46 | } |
| 47 |
no test coverage detected