Home
last modified time | relevance | path

Searched refs:ReuseKernelNodeInfo (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/fl/server/kernel/
Daggregation_kernel.h79 …const 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_;
Doptimizer_kernel.h62 …const 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_;
/third_party/mindspore/mindspore/ccsrc/fl/server/
Dparameter_aggregator.cc212 …const ReuseKernelNodeInfo &reuse_kernel_node_inputs_info = aggr_kernel->reuse_kernel_node_inputs_i… in InitAggregationKernels()
242 …const ReuseKernelNodeInfo &reuse_kernel_node_inputs_info = optimizer_kernel->reuse_kernel_node_inp… in InitOptimizerKernels()
257 const ReuseKernelNodeInfo &reuse_kernel_node_inputs_info, in AssignMemory()
389 … const ReuseKernelNodeInfo &reuse_kernel_node_inputs_info,
394 … const ReuseKernelNodeInfo &reuse_kernel_node_inputs_info,
Dparameter_aggregator.h115 const ReuseKernelNodeInfo &reuse_kernel_node_inputs_info,
Dcommon.h100 using ReuseKernelNodeInfo = std::map<std::string, size_t>; variable