Searched refs:reuse_kernel_node_inputs_info (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/fl/server/ |
| D | parameter_aggregator.cc | 212 …const ReuseKernelNodeInfo &reuse_kernel_node_inputs_info = aggr_kernel->reuse_kernel_node_inputs_i… in InitAggregationKernels() local
213 if (!AssignMemory(aggr_kernel, cnode, reuse_kernel_node_inputs_info, memory_register_)) { in InitAggregationKernels()
242 …const ReuseKernelNodeInfo &reuse_kernel_node_inputs_info = optimizer_kernel->reuse_kernel_node_inp… in InitOptimizerKernels() local
243 if (!AssignMemory(optimizer_kernel, cnode, reuse_kernel_node_inputs_info, memory_register_)) { in InitOptimizerKernels()
257 const ReuseKernelNodeInfo &reuse_kernel_node_inputs_info, in AssignMemory() argument
272 if (reuse_kernel_node_inputs_info.size() > input_names.size()) { in AssignMemory()
274 … << reuse_kernel_node_inputs_info.size() << ", but input_names size is " << input_names.size(); in AssignMemory()
284 if (reuse_kernel_node_inputs_info.count(name) != 0) { in AssignMemory()
287 size_t index = reuse_kernel_node_inputs_info.at(name); in AssignMemory()
389 … const ReuseKernelNodeInfo &reuse_kernel_node_inputs_info,
[all …]
|
| D | parameter_aggregator.h | 115 const ReuseKernelNodeInfo &reuse_kernel_node_inputs_info,
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/kernel/ |
| D | aggregation_kernel.h | 79 …const ReuseKernelNodeInfo &reuse_kernel_node_inputs_info() { return reuse_kernel_node_inputs_info_… in reuse_kernel_node_inputs_info() function
|
| D | optimizer_kernel.h | 62 …const ReuseKernelNodeInfo &reuse_kernel_node_inputs_info() { return reuse_kernel_node_inputs_info_… in reuse_kernel_node_inputs_info() function
|