| 124 | } |
| 125 | |
| 126 | struct CPUNDAlloc { |
| 127 | void AllocData(DLTensor* tensor) { tensor->data = malloc(tvm::ffi::GetDataSize(*tensor)); } |
| 128 | void FreeData(DLTensor* tensor) { free(tensor->data); } |
| 129 | }; |
| 130 | |
| 131 | void ExampleTensor() { |
| 132 | namespace ffi = tvm::ffi; |
no outgoing calls