hlu1 commented on issue #5381: URL: https://github.com/apache/incubator-tvm/pull/5381#issuecomment-617442169
I don't think you need to implement a non-copy version of set_output if you simply want to avoid copy. `GetOutput` returns a NDArray, which is essentially a shared_ptr to the underlying tensor: https://github.com/apache/incubator-tvm/blob/master/src/runtime/graph/graph_runtime.cc#L156-L160 ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected]
