Home
last modified time | relevance | path

Searched refs:aggr_kernel (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/fl/server/
Dparameter_aggregator.cc103 std::shared_ptr<kernel::AggregationKernel> aggr_kernel = aggregator_with_params.first; in LaunchAggregators() local
104 MS_ERROR_IF_NULL_W_RET_VAL(aggr_kernel, false); in LaunchAggregators()
105 bool ret = aggr_kernel->Launch(params.inputs, params.workspace, params.outputs); in LaunchAggregators()
107 …MS_LOG(ERROR) << "Launching aggregation kernel " << typeid(aggr_kernel.get()).name() << " failed."; in LaunchAggregators()
159 std::shared_ptr<kernel::AggregationKernel> aggr_kernel = aggregator_with_params.first; in ResetAggregationStatus() local
160 if (aggr_kernel == nullptr) { in ResetAggregationStatus()
164 aggr_kernel->Reset(); in ResetAggregationStatus()
179 std::shared_ptr<kernel::AggregationKernel> aggr_kernel = aggregator_with_params.first; in IsAggregationDone() local
180 MS_ERROR_IF_NULL_W_RET_VAL(aggr_kernel, false); in IsAggregationDone()
181 if (!aggr_kernel->IsAggregationDone()) { in IsAggregationDone()
[all …]
Dparameter_aggregator.h120 …bool GenerateAggregationKernelParams(const std::shared_ptr<kernel::AggregationKernel> &aggr_kernel,