facebookarchive/caffe2

MKL-DNN integration does not support backward pass

Open

#2.376 aberto em 22 de mar. de 2018

Ver no GitHub
 (1 comment) (0 reactions) (0 assignees)Shell (1.954 forks)batch import
help wanted

Métricas do repositório

Stars
 (8.424 stars)
Métricas de merge de PR
 (Nenhuma PRs mesclada em 30d)

Description

MKL-DNN integration (in caffe2/mkl) implements only forward pass operators, although MKL-DNN itself provides primitives for both forward and backward pass. Implementation of ConvGradient operator would let Caffe2 use MKL-DNN for training on CPU, and support some models which need backprop (e.g. deep dream).

Guia do colaborador