Commit Graph

19 Commits (f1a9efcac5743e05fc4e1dbc53c08f69ec19d5a8)

Author SHA1 Message Date
QI JUN 61ec0b9516
Refine device context (#6433)
7 years ago
chengduoZH 5238c9fb0b input type should be different
7 years ago
chengduoZH 2160497786 fix pooling functor parameter order
7 years ago
chengduoZH 36da82550a Add code comments
7 years ago
chengduoZH e21e5646a5 fix atomicAdd -> CudaAtomicAdd
7 years ago
chengduoZH bb33c2b3a5 fix kernel func
8 years ago
chengduoZH 2ed56df1e6 remove conflict
8 years ago
chengduoZH bee95fc891 fix code format and some bug
8 years ago
chengduoZH e1e3859e88 remove custom attr checker and fix code format
8 years ago
chengduoZH 6326c40d27 Add max pool with index
8 years ago
chengduoZH dfc8d3c1c1 Fix (According to the review)
8 years ago
chengduoZH 905a462d3c Merge branch 'fix_maxpool_backward_functor_temp' into Add_pool_op_temp
8 years ago
chengduoZH b72854389e Fix (According to the review)
8 years ago
chengduoZH 9b1431b8d9 fix maxpool backward functor and add unit test
8 years ago
chengduoZH b0a47b149d remove CHECK in *.cu
8 years ago
chengduoZH 84a2512b90 fix parameter name and function define
8 years ago
chengduoZH 3416f5e0f8 fix function define
8 years ago
chengduoZH a8477381ac Add pooling2d(max, ave) and pooling3d(max, ave) Op
8 years ago
chengduoZH 9e7c0b5ef7 Add pooling2d(max, ave) and pooling3d(max, ave) functor
8 years ago