Assume I have a operator, I wanna it’s input and output use the same memory. Could It be done with TVM? And How?
I wanna use this feature with te.exterm
so that I cound use my custom operator without output memory allocation.
Assume I have a operator, I wanna it’s input and output use the same memory. Could It be done with TVM? And How?
I wanna use this feature with te.exterm
so that I cound use my custom operator without output memory allocation.
I think it’s possible. Try not passing any output tensor to te.extern
, and directly modify the input tensor.