You need to sign in or sign up before continuing.
Created by: tensor-tang
-
enable
MKLDNNMatrix
copy fromCpuMatrix
, then theMKLDNNMatrix
mean and var can copy from moving mean and var. -
add
MKLDNNBatchNormLayer
files -
add unit test for
mkldnn_batch_norm
layer -
add python interface for
mkldnn_batch_norm
type. BTW, find a question that the comment saiduse_cudnn
should depends oncudnn_version
, but it has never been used. Maybe forget addcudnn_version >= 4007
or been removed? @luotao1 -
add unit test for
mkldnn_batch_norm
branch test and simple net test