max_pool2d already has InferCorrectLayouts
, so it’s possible that your ConvertOpLayout
is not effective. You may try to remove the InferCorrectLayouts in src/relay/nn/pooling.cc
to verify if it is the case.
Also cc @anijain2305
max_pool2d already has InferCorrectLayouts
, so it’s possible that your ConvertOpLayout
is not effective. You may try to remove the InferCorrectLayouts in src/relay/nn/pooling.cc
to verify if it is the case.
Also cc @anijain2305