Commit Graph

21 Commits (af215a1a532137686a696cb1c5da5a8797ac51ca)

Author SHA1 Message Date
tensor-tang 698071cc33 share mkldnn output value data if next layer is cpu device
7 years ago
tensor-tang 6715beaace enable merge output grad of mkldnn
7 years ago
Luo Tao c53d214205 fix compiler warning from MKLDNNLayer and so on
7 years ago
tensor-tang ed27a3be78 add some log info
8 years ago
tensor-tang 4ec3a77a43 should run resetBwd before bwdAct
8 years ago
tensor-tang 49d4b39f28 fix typo and remove some unused code
8 years ago
tensor-tang b0d9b68a5f unify functions of mkldnn_fc and refine comments
8 years ago
tensor-tang 4f0869bef3 add some necessary params for reset functions
8 years ago
tensor-tang 94ea8ee0e5 refine MKLDNNLayer logical: move forward and backward to MKLDNNLayer and remove copyOutputInfoToOtherDevice
8 years ago
tensor-tang f40d5f580d remove syncOutputGrad, rename syncInputValue to updateInputData
8 years ago
tensor-tang d4c0734840 remove convertOutputToOtherDevice
8 years ago
tensor-tang c5183caa04 rename
8 years ago
tensor-tang bfbd066fdd refine
8 years ago
tensor-tang 98b7c67364 add todo
8 years ago
tensor-tang 4cc57836f3 enable reorder
8 years ago
tensor-tang 780c8d969e make downSpatial work, and remove hasSpatial_
8 years ago
tensor-tang 48d87e5e91 pass test, support input CPU device
8 years ago
tensor-tang 4eecd0c2d5 use MKLDNNMatrix in fc backward
8 years ago
tensor-tang 4bffbd30f0 use MKLDNNMatrix in fc forward
8 years ago
tensor-tang e08651f9b5 remove flag use_mkldnn_wgt
8 years ago
tensor-tang a475a57d9b rename files and classes, use uppercase of Mkldnn and Cpu
8 years ago