Commit Graph

32 Commits (bc0d2557969bade8a3049850798e9eba8505e968)

Author SHA1 Message Date
tensor-tang bc0d255796 make MKLDNNLayer input value as a vector
7 years ago
tensor-tang 3117d97784 add inputChannel in resetInValue for concat layer
7 years ago
tensor-tang c397599dfd remove weight and bias in MKLDNN reset function, since not all layers have weight and bias.
7 years ago
tensor-tang a9490a1053 make output channels changeable in reshape function
7 years ago
tensor-tang 9acfba82a3 add input index choice for mkldnn_concat
7 years ago
tensor-tang 2a77418668 refine reset input buffers, make it support more than one input.
8 years ago
tensor-tang 386bb1fece fix crash on ds2
8 years ago
tensor-tang 5c892db64c remove unused code
8 years ago
tensor-tang d75b00c221 refine the gtest log info and vlog order,
8 years ago
tensor-tang 94e442d4b1 add cpp file of MKLDNNLayer
8 years ago
tensor-tang 9e38dafa29 change MKLDNNMatrix create interface since MatrixPtr is not always required
8 years ago
tensor-tang c1914543b0 refine mkldnn logic, move reset buffers into MKLDNNLayer
8 years ago
tensor-tang 1199aa6876 fix bug: clear grad and always share data when output has cpu
8 years ago
tensor-tang 7a7c8fd9e5 simplify some comments and code
8 years ago
tensor-tang e1954857d7 fix bug: merge grad must before backward act.
8 years ago
tensor-tang 698071cc33 share mkldnn output value data if next layer is cpu device
8 years ago
tensor-tang 6715beaace enable merge output grad of mkldnn
8 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 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 d74fe78040 refine MKLDNNMatrix, solid data handle, rename updateData to setData
8 years ago
tensor-tang c5183caa04 rename
8 years ago
tensor-tang bfbd066fdd refine
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 62e6dac402 add MKLDNNMatrix files
8 years ago
tensor-tang a475a57d9b rename files and classes, use uppercase of Mkldnn and Cpu
8 years ago