e445b3ff20
Move framework.proto to proto namespace ( #6718 )
...
* Move framework.proto to proto namespace
* Fix compile
* Fix compile
* Fix Compile
7 years ago
61ec0b9516
Refine device context ( #6433 )
...
There are mainly following fixes:
- take `DeviceContext` as the template parameter of math functors and OpKernel instead of `Place`
- remove `eigen_device` interface in base class `DeviceContext`
- remove `GetEigenDevice` interface in `ExecutionContext` and base class `DeviceContext`
- remove unused `platform::EigenDeviceConverter`
- rename `REGISTER_OP_GPU_KERNEL` to `REGISTER_OP_CUDA_KERNEL`
- rename `USE_GPU_ONLY_OP` to `USE_CUDA_ONLY_OP`
7 years ago
2160497786
fix pooling functor parameter order
7 years ago
cdf5e87104
fix attr name
8 years ago
6bdf5c1417
fix bug
8 years ago
0ab012cf7f
fix doc
8 years ago
73a8b78a72
Correct OpWithKernel's infershape ( #4847 )
...
They are public now
8 years ago
75d0c79015
Change Name convention of operator attributes ( #4807 )
...
* Change dataType to data_type
Follow PEP8
* Change name_convention to fit PEP8
8 years ago
6db476ed89
Separate the declarations and implementation of the PoolOp and PoolMaker class in order to reuse in pool_cudnn
8 years ago
df59889984
remove conflict
8 years ago
e1e3859e88
remove custom attr checker and fix code format
8 years ago
4b06d8db91
fix globalPooling type (int => bool)
8 years ago
dfc8d3c1c1
Fix (According to the review)
8 years ago
f6e69d7412
fix maxpool backward functor
8 years ago
b72854389e
Fix (According to the review)
8 years ago
0417e4e4bf
fix framework::LoDTensor => Tensor
8 years ago
84a2512b90
fix parameter name and function define
8 years ago
3416f5e0f8
fix function define
8 years ago
a8477381ac
Add pooling2d(max, ave) and pooling3d(max, ave) Op
8 years ago
Yu Yang
QI JUN
chengduoZH