• Q
    Refine device context (#6433) · 61ec0b95
    QI JUN 提交于
    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`
    61ec0b95
pooling.cu 44.8 KB