Searched refs:aggr_params (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/fl/server/ |
| D | parameter_aggregator.cc | 307 KernelParams aggr_params = {}; in GenerateAggregationKernelParams() local
310 …(void)std::transform(input_names.begin(), input_names.end(), std::back_inserter(aggr_params.inputs… in GenerateAggregationKernelParams()
314 …ransform(workspace_names.begin(), workspace_names.end(), std::back_inserter(aggr_params.workspace), in GenerateAggregationKernelParams()
318 …(void)std::transform(output_names.begin(), output_names.end(), std::back_inserter(aggr_params.outp… in GenerateAggregationKernelParams()
321 aggr_kernel->SetParameterAddress(aggr_params.inputs, aggr_params.workspace, aggr_params.outputs); in GenerateAggregationKernelParams()
322 aggregation_kernel_parameters_.push_back(std::make_pair(aggr_kernel, aggr_params)); in GenerateAggregationKernelParams()
|