May I ask where is the code that corresponds to the input buffer and weight buffer sections?
Because I want to optimize this part of the code. For exampl, I want to change TVM where computed data structures (e.g., instructions, microoperations, weights) are reused in subsequent activations. Is this possible?