| 54 | } |
| 55 | |
| 56 | bool get_tensor_memptr(const Tensor* tensor, TensorMemPtr& ptr) |
| 57 | { |
| 58 | if(tensor->ExistAttr("tensor_mem")) |
| 59 | { |
| 60 | ptr = any_cast<TensorMemPtr>(tensor->GetAttr("tensor_mem")); |
| 61 | return true; |
| 62 | } |
| 63 | |
| 64 | return false; |
| 65 | } |
| 66 | |
| 67 | bool set_tensor_mem(Tensor* tensor, void* addr, int size, mem_release_t releaser) |
| 68 | { |
no test coverage detected