Code

im2col based conv2d#802

Code

Merged

LaurentMazare merged 6 commits intomainhuggingface/candle:mainfrom conv2d-im2colhuggingface/candle:conv2d-im2colCopy head branch name to clipboardSep 10, 2023

+210-16

Commits

Commits on Sep 10, 2023

• im2col implementation for conv2d.

  LaurentMazare

  committed
  778b041

  Copy full SHA for 778b041

• Fix for the im2col implementation to match the current conv2d.

  LaurentMazare

  committed
  bf2fc77

  Copy full SHA for bf2fc77

• Small optimization.

  LaurentMazare

  committed
  5cc7d4c

  Copy full SHA for 5cc7d4c

• Add a cuda kernel.

  LaurentMazare

  committed
  57b39bb

  Copy full SHA for 57b39bb

• Handle arbitrary layouts.

  LaurentMazare

  committed
  177434a

  Copy full SHA for 177434a

• Im2Col cuda code.

  LaurentMazare

  committed
  69ba85a

  Copy full SHA for 69ba85a