• 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
math_function_test.cc 4.8 KB
反馈
建议
客服 返回
顶部