lhutton1 commented on pull request #6430: URL: https://github.com/apache/incubator-tvm/pull/6430#issuecomment-690584984
So since convert layout is part of preprocessing the module in the optimize step before codegen (https://github.com/apache/incubator-tvm/blob/master/src/relay/backend/contrib/arm_compute_lib/codegen.cc#L342) we could do this: ``` with TempOpAttr("nn.conv2d", "FTVMConvertOpLayout", ...): graph, lib, parmas = tvm.build(...) ``` However it's not good expect the user to know how to add this. Therefore, my plan was to set the attribute for the scope of `PreProcessModule` https://github.com/apache/incubator-tvm/blob/master/src/relay/backend/contrib/arm_compute_lib/codegen.cc#L340. Hope that makes more sense? ---------------------------------------------------------------- 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]
