| 131 | } |
| 132 | |
| 133 | static void MGBTensor2Tensor(const MGBTensor* mgb_tensor, Tensor* tensor) { |
| 134 | tensor->layout.nr_dim = mgb_tensor->layout.shape.ndim; |
| 135 | for (int i = 0; i < mgb_tensor->layout.shape.ndim; ++i) { |
| 136 | tensor->layout.dims[i] = mgb_tensor->layout.shape.shape[i]; |
| 137 | } |
| 138 | |
| 139 | switch (mgb_tensor->layout.dtype) { |
| 140 | case MGB_DTYPE_FLOAT32: |
| 141 | tensor->dtype.type_enum = TinyNN_FLOAT; |
| 142 | break; |
| 143 | case MGB_DTYPE_FLOAT16: |
| 144 | tensor->dtype.type_enum = TinyNN_FLOAT16; |
| 145 | break; |
| 146 | case MGB_DTYPE_INT32: |
| 147 | tensor->dtype.type_enum = TinyNN_INT; |
| 148 | break; |
| 149 | case MGB_DTYPE_INT16: |
| 150 | tensor->dtype.type_enum = TinyNN_INT16; |
| 151 | break; |
| 152 | case MGB_DTYPE_UINT8: |
| 153 | tensor->dtype.type_enum = TinyNN_UINT8; |
| 154 | break; |
| 155 | default: |
| 156 | TINYNN_ASSERT_MSG(0, "Unsupport data type\n"); |
| 157 | } |
| 158 | } |
| 159 | |
| 160 | #if defined(_WIN32) |
| 161 | #include <io.h> |