paddle/fluid/operators/optimizers/adam_op_npu.cc · 781f40282599118361627b04cd3d956552c7a397 · SummerGao. / Paddle

由 Leo Chen 提交于 7月 22, 2021

* copy found_inf to cpu in advance to improve performance

* add npu test

* add npu test

* refine code

* refine memcpy op

* fix adam

781f4028

adam_op_npu.cc 9.5 KB

SummerGao. / Paddle 与 Fork 源项目一致