Searched refs:params_info (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/fl/server/kernel/ |
| D | kernel_factory.h | 56 void Register(const std::string &name, const ParamsInfo ¶ms_info, C &&creator) { in Register() argument
57 name_to_creator_map_[name].push_back(std::make_pair(params_info, creator)); in Register()
66 const ParamsInfo ¶ms_info = name_type_creator.first; in Create() local
68 if (Matched(params_info, kernel_node)) { in Create()
70 kernel->set_params_info(params_info); in Create()
82 virtual bool Matched(const ParamsInfo ¶ms_info, const CNodePtr &kernel_node) { return true; } in Matched() argument
|
| D | optimizer_kernel_factory.cc | 24 bool OptimizerKernelFactory::Matched(const ParamsInfo ¶ms_info, const CNodePtr &kernel_node) { in Matched() argument
33 size_t input_num = params_info.inputs_num(); in Matched()
35 auto one_input_name_type = params_info.inputs_name_type(i); in Matched()
50 size_t output_num = params_info.outputs_num(); in Matched()
52 auto one_output_name_type = params_info.outputs_name_type(i); in Matched()
|
| D | aggregation_kernel_factory.cc | 24 bool AggregationKernelFactory::Matched(const ParamsInfo ¶ms_info, const CNodePtr &kernel_node) { in Matched() argument
33 size_t input_num = params_info.inputs_num(); in Matched()
35 auto one_input_name_type = params_info.inputs_name_type(i); in Matched()
51 size_t output_num = params_info.outputs_num(); in Matched()
53 auto one_output_name_type = params_info.outputs_name_type(i); in Matched()
|
| D | optimizer_kernel_factory.h | 45 bool Matched(const ParamsInfo ¶ms_info, const CNodePtr &kernel_node) override;
50 …OptimizerKernelRegister(const std::string &name, const ParamsInfo ¶ms_info, OptimizerKernelCre… in OptimizerKernelRegister() argument
51 OptimizerKernelFactory::GetInstance().Register(name, params_info, std::move(creator)); in OptimizerKernelRegister()
|
| D | aggregation_kernel_factory.h | 45 bool Matched(const ParamsInfo ¶ms_info, const CNodePtr &kernel_node) override;
50 AggregationKernelRegister(const std::string &name, const ParamsInfo ¶ms_info, in AggregationKernelRegister() argument
52 AggregationKernelFactory::GetInstance().Register(name, params_info, std::move(creator)); in AggregationKernelRegister()
|
| D | aggregation_kernel.h | 73 void set_params_info(const ParamsInfo ¶ms_info) { params_info_ = params_info; } in set_params_info() argument
|
| D | optimizer_kernel.h | 56 void set_params_info(const ParamsInfo ¶ms_info) { params_info_ = params_info; } in set_params_info() argument
|
| /third_party/mindspore/mindspore/ccsrc/fl/ |
| D | CMakeLists.txt | 24 list(REMOVE_ITEM _FL_SRC_FILES "server/kernel/params_info.cc")
|