We are pleased to share the codes. Please check the PR: [TOPI][Tensor Core] Conv2d and Dense ops support on Tensor Core #5099. Try to find the code in topi/python/topi/cuda/conv2d_nhwc.py, which is the code that has the same layout as conv2d of Tensor Core.
For any questions, please feel free to let me know. --- [Visit Topic](https://discuss.tvm.ai/t/rfc-tensor-core-optimization-of-cnns-on-tensor-core/6004/19) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.ai/email/unsubscribe/cb45d9973c538a056abf11b74b19d50aa047c0357da7c92a6a0f1e77ad2c3b4b).