| 39 | */ |
| 40 | template <typename T, typename... Ts> |
| 41 | inline Strides compute_strides(const ITensorInfo &info, T stride_x, Ts &&...fixed_strides) |
| 42 | { |
| 43 | const TensorShape &shape = info.tensor_shape(); |
| 44 | |
| 45 | // Create strides object |
| 46 | Strides strides(stride_x, fixed_strides...); |
| 47 | |
| 48 | for (size_t i = 1 + sizeof...(Ts); i < info.num_dimensions(); ++i) |
| 49 | { |
| 50 | strides.set(i, shape[i - 1] * strides[i - 1]); |
| 51 | } |
| 52 | |
| 53 | return strides; |
| 54 | } |
| 55 | |
| 56 | /** Create a strides object based on the tensor dimensions. |
| 57 | * |
no test coverage detected