mindspore2022/mindspore/ccsrc/fl/server/kernel
tronzhang 27893cde7c change kernelmod interface and gpu/cpu kernel mod 2022-04-08 16:28:02 +08:00
..
round !31368 fl compression 2022-03-22 03:17:43 +00:00
aggregation_kernel.h change kernelmod interface and gpu/cpu kernel mod 2022-04-08 16:28:02 +08:00
aggregation_kernel_factory.cc extract common as an independent shared library 2022-02-26 10:55:51 +08:00
aggregation_kernel_factory.h refactor kernel mod class and subclass 2022-01-22 16:31:37 +08:00
apply_momentum_kernel.cc Fix fl namespace issue. 2021-07-08 21:52:21 +08:00
apply_momentum_kernel.h Refactor ccsrc directories and CMakeLists files 2022-02-13 14:37:51 +08:00
dense_grad_accum_kernel.cc Fix fl namespace issue. 2021-07-08 21:52:21 +08:00
dense_grad_accum_kernel.h FL, opt allreduce 2022-03-01 15:16:24 +08:00
fed_avg_kernel.cc Fix fl namespace issue. 2021-07-08 21:52:21 +08:00
fed_avg_kernel.h fix issue I4TBJO 2022-03-09 10:52:18 +08:00
kernel_factory.h code sync in federated learning 2021-12-02 14:41:56 +08:00
optimizer_kernel.h change kernelmod interface and gpu/cpu kernel mod 2022-04-08 16:28:02 +08:00
optimizer_kernel_factory.cc code sync in federated learning 2021-12-02 14:41:56 +08:00
optimizer_kernel_factory.h refactor kernel mod class and subclass 2022-01-22 16:31:37 +08:00
params_info.cc Fix fl namespace issue. 2021-07-08 21:52:21 +08:00
params_info.h Fix fl namespace issue. 2021-07-08 21:52:21 +08:00
sgd_kernel.cc refactor kernel mod class and subclass 2022-01-22 16:31:37 +08:00
sgd_kernel.h refactor cpu factory 2022-03-14 10:05:56 +08:00