Commit Graph

33 Commits (f086ebb8b9b8e21f7d0d2056f6a961da080edfb2)

Author SHA1 Message Date
chengduoZH c9641a03dc refine code
7 years ago
chengduoZH ed7e74abf2 follow comments and refine python doc
7 years ago
chengduoZH 24f528a1a5 follow comments
7 years ago
chengduoZH 251c6032fb set use_cudnn as default
7 years ago
chengduoZH 79aa51229a fix conv, pool, conv_trans to decide use cudnn or not
7 years ago
dzhwinter 5ad1aef051
"cudnn operators change to cudnn kernel" (#6660)
7 years ago
Yang Yu 040dc59b0f Correctly handle image operators
7 years ago
Yu Yang e445b3ff20
Move framework.proto to proto namespace (#6718)
7 years ago
QI JUN 61ec0b9516
Refine device context (#6433)
7 years ago
chengduoZH 9abc0e04c1 fix conv and conv_trans op doc
7 years ago
chengduoZH 3bd3cc0c85 add double type for kernel
7 years ago
chengduoZH cdf5e87104 fix attr name
7 years ago
kexinzhao 5d8cdf2031 Polish operator docs (n to p) (#5376)
7 years ago
chengduoZH 6bdf5c1417 fix bug
7 years ago
chengduoZH cadee843b8 follow comments
7 years ago
chengduoZH 0ab012cf7f fix doc
7 years ago
Yu Yang 75d0c79015 Change Name convention of operator attributes (#4807)
7 years ago
chengduoZH 4849fba7e9 follow comments
7 years ago
chengduoZH 6db476ed89 Separate the declarations and implementation of the PoolOp and PoolMaker class in order to reuse in pool_cudnn
7 years ago
chengduoZH 67edd04a2f fix doc
7 years ago
qiaolongfei c0a34e1c64 rename InferShapeContextBase to InferShapeContext
7 years ago
chengduoZH e1e3859e88 remove custom attr checker and fix code format
7 years ago
chengduoZH 3c0f079333 remove conflict and fix InferShape function
7 years ago
chengduoZH 4b06d8db91 fix globalPooling type (int => bool)
7 years ago
chengduoZH dfc8d3c1c1 Fix (According to the review)
7 years ago
chengduoZH 8c478b36e2 fix Atrr check
7 years ago
chengduoZH b72854389e Fix (According to the review)
7 years ago
chengduoZH f2ccc11f59 fix pool doc (pool_op.cc)
7 years ago
chengduoZH 0417e4e4bf fix framework::LoDTensor => Tensor
7 years ago
chengduoZH 6f61b5df7d fix unit test
7 years ago
chengduoZH 84a2512b90 fix parameter name and function define
7 years ago
chengduoZH 3416f5e0f8 fix function define
7 years ago
chengduoZH a8477381ac Add pooling2d(max, ave) and pooling3d(max, ave) Op
8 years ago