v0.9.dev0
Lyken17
tagged this
12 Nov 17:05
…upport groups (#9465) * 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>
Assets 2
-
2021-11-12T17:05:39Z -
2021-11-12T17:05:39Z -