I follow tune arm cpu tutorial on my custom TF model (imported like described here) and receive the following error:
AssertionError: only support NCHW currently
Is there any workaround for that?
After changing layout = None like in tutorial to layout = “NCHW” autotuning runs