| 80 | } |
| 81 | |
| 82 | int LITE_set_tensor_layout(LiteTensor tensor, const LiteLayout layout) { |
| 83 | LITE_CAPI_BEGIN(); |
| 84 | LITE_ASSERT(tensor, "The tensor pass to LITE c_api is null"); |
| 85 | auto tensor_ptr = static_cast<lite::Tensor*>(tensor); |
| 86 | tensor_ptr->set_layout(convert_to_layout(layout)); |
| 87 | LITE_CAPI_END(); |
| 88 | } |
| 89 | |
| 90 | int LITE_reset_tensor_memory( |
| 91 | LiteTensor tensor, void* prepared_data, size_t data_length_in_byte) { |