Giters

libxsmm / libxsmm-dnn

Reference implementation of Deep Neural Network primitives using LIBXSMM's Tensor Processing Primitives (TPP)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

The Convolution layout Issue

wpy1368064015 opened this issue · comments

commented

Can libxsmm support the NCHW and KCRS layout for convolution?

commented

There are various routines to copy whatever format into or out of LIBXSMM-DNN's internal formats (NCHWc, KCRSck). Routines exist for all supported type-flavors (as well as modded versions omitting certain transposes).

commented

Do you have something specifically in mind with this question, e.g., sample code?

commented

I improved getting started with LIBXSMM-DNN and added a README. Though, we just recently transitioned this code base from LIBXSMM into a separate/this repository.

Thank you for your question, and feel free to raise any issue you might run into!