[GitHub] [tvm] Alex-grovety edited a comment on issue #8717: Fusion of operations and cast in mobilenet v1 conv2d causing large feature maps

[GitBox <gi...@apache.org>]

Alex-grovety edited a comment on issue #8717:
URL: https://github.com/apache/tvm/issues/8717#issuecomment-921856470


   If I understand correctly it will be
   ```
   elif layout == "NHWC":
       assert kernel_layout == "HWOI"
       is_aarch64 = topi.arm_cpu.arm_utils.is_aarch64_arm()
       if is_aarch64 or "+neon" in target.mattr:
           strategy.add_implementation(
               wrap_compute_conv2d(topi.arm_cpu.compute_depthwise_conv2d_nhwc),
               wrap_topi_schedule(topi.arm_cpu.schedule_depthwise_conv2d_nhwc),
               name="depthwise_conv2d_nhwc.arm_cpu",
           )
       else:
           strategy.add_implementation(
               wrap_compute_conv2d(topi.nn.depthwise_conv2d_nhwc),
               wrap_topi_schedule(topi.x86.schedule_depthwise_conv2d_nhwc),
               name="depthwise_conv2d_nhwc.arm_cpu",
           )
   ```
   instead of
   https://github.com/apache/tvm/blob/02fbaf0ed9120a8f95155e63de42459f230584aa/python/tvm/relay/op/strategy/arm_cpu.py#L198-L204
   


-- 
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.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org