github-featured / apache--tvm

发布

0 发布 63 Git 标签

RSS 订阅源
  • v0.9.dev0 3ad7c4a769
    比较

    [Conv2DTransposed] Fix wrong shape check and add new TOPI module to support groups (#9465)

    frostbyte_neo 发布于 2021-11-12 17:05:39 +00:00

    • f wrong type check in conv2d_transpose

    • add test case for conv2d transpose

    • add groups support for conv2d_transpose

    • add naive implementation and schedule for conv2d with groups

    • enable tests for cpu and arm_cpu, raise error for cuda platform

    • revert the cuda and generic strategy

    • revert back the x86 strategy

    • revert back the arm_cpu strategy

    • revert back the arm_cpu strategy

    • revert back the arm_cpu strategy

    • fix EOF of x86

    • clang lint updated c++ code

    • update topi implementation

    • Revert test

    • Revert test

    • add generic/x86/arm specialization for conv2d_transpose with groups > 1

    • remove commentted codes

    • fix lint

    • fix lint

    • fix c++ lint

    • fix lint

    • fix python lint

    • remove comments and reformat

    • lint file

    • lint code

    • fix lint

    • update logging information in convolution.h

    Co-authored-by: Alicja Kwasniewska alicja.kwasniewska@sima.ai

    下载附件