Searched refs:reuse_kernel_node_inputs_info_ (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/fl/server/kernel/ |
| D | sgd_kernel.h | 51 reuse_kernel_node_inputs_info_.insert(std::make_pair(kWeight, 0)); in GenerateReuseKernelNodeInfo()
52 reuse_kernel_node_inputs_info_.insert(std::make_pair(kLearningRate, 2)); in GenerateReuseKernelNodeInfo()
53 reuse_kernel_node_inputs_info_.insert(std::make_pair(kAccumulation, 3)); in GenerateReuseKernelNodeInfo()
54 reuse_kernel_node_inputs_info_.insert(std::make_pair(kMomentum, 4)); in GenerateReuseKernelNodeInfo()
55 reuse_kernel_node_inputs_info_.insert(std::make_pair(kStat, 5)); in GenerateReuseKernelNodeInfo()
|
| D | apply_momentum_kernel.h | 52 reuse_kernel_node_inputs_info_.insert(std::make_pair(kWeight, 0)); in GenerateReuseKernelNodeInfo()
53 reuse_kernel_node_inputs_info_.insert(std::make_pair(kAccumulation, 1)); in GenerateReuseKernelNodeInfo()
54 reuse_kernel_node_inputs_info_.insert(std::make_pair(kLearningRate, 2)); in GenerateReuseKernelNodeInfo()
55 reuse_kernel_node_inputs_info_.insert(std::make_pair(kMomentum, 4)); in GenerateReuseKernelNodeInfo()
|
| D | aggregation_kernel.h | 79 …nst ReuseKernelNodeInfo &reuse_kernel_node_inputs_info() { return reuse_kernel_node_inputs_info_; } in reuse_kernel_node_inputs_info()
100 ReuseKernelNodeInfo reuse_kernel_node_inputs_info_; variable
|
| D | optimizer_kernel.h | 62 …nst ReuseKernelNodeInfo &reuse_kernel_node_inputs_info() { return reuse_kernel_node_inputs_info_; } in reuse_kernel_node_inputs_info()
91 ReuseKernelNodeInfo reuse_kernel_node_inputs_info_; variable
|
| D | fed_avg_kernel.h | 194 (void)reuse_kernel_node_inputs_info_.insert(std::make_pair(kWeight, cnode_weight_idx_)); in GenerateReuseKernelNodeInfo()
|