| 116 | } |
| 117 | |
| 118 | inline bool FastTransposeCopyValid(const DenseTensor& self, |
| 119 | const DenseTensor& src) { |
| 120 | constexpr int64_t MIN_NUMEL = 360; |
| 121 | return src.numel() != 0 && src.dims().size() == 2 && src.strides()[0] == 1 && |
| 122 | src.strides()[1] == src.dims()[0] && |
| 123 | self.dims().size() == src.dims().size() && self.numel() >= MIN_NUMEL; |
| 124 | } |
| 125 | |
| 126 | template <typename T, typename Context> |
| 127 | void ContiguousKernel(const Context& dev_ctx, |
no test coverage detected