| 31 | } |
| 32 | |
| 33 | inline void DealWithStride(const DenseTensorIterator& iter, int64_t* strides) { |
| 34 | for (int dim = 0; dim < iter.ndim(); dim++) { |
| 35 | for (int arg = 0; arg < iter.ntensors(); arg++) { |
| 36 | *strides++ = iter.strides(arg)[dim]; |
| 37 | } |
| 38 | } |
| 39 | if (iter.ndim() < 2) { |
| 40 | std::fill_n(strides, (2 - iter.ndim()) * iter.ntensors(), 0); |
| 41 | } |
| 42 | } |
| 43 | |
| 44 | template <typename T> |
| 45 | inline void FallbackContiguous(const DDim& input_dims, |
no test coverage detected