Commit Graph

24 Commits (2dff98ca11a48afcceedbfb4ec6ead4eddff0118)

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