facebookarchive/caffe2
Ver no GitHubMKL-DNN integration does not support backward pass
Open
#2.376 aberto em 22 de mar. de 2018
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).