| 33 | } |
| 34 | |
| 35 | inline Tensor EmptyStrided(const Shape& shape, const Shape& strides, DLDataType dtype, |
| 36 | DLDevice device) { |
| 37 | return Tensor::FromNDAllocStrided(CPUNDAlloc(), shape, strides, dtype, device); |
| 38 | } |
| 39 | |
| 40 | int TestEnvTensorAllocator(DLTensor* prototype, TVMFFIObjectHandle* out) { |
| 41 | Shape shape(prototype->shape, prototype->shape + prototype->ndim); |