| 98 | } |
| 99 | |
| 100 | void free_tensor_mem(Tensor* tensor) |
| 101 | { |
| 102 | if(tensor->GetType() == kConstTensor) |
| 103 | return; |
| 104 | |
| 105 | if(tensor->ExistAttr("tensor_mem")) |
| 106 | tensor->RemoveAttr("tensor_mem"); |
| 107 | } |
| 108 | |
| 109 | } // namespace TEngine |
no test coverage detected