1 /**
2 * Copyright 2021 Huawei Technologies Co., Ltd
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include "fl/server/kernel/params_info.h"
18 #include "utils/log_adapter.h"
19
20 namespace mindspore {
21 namespace fl {
22 namespace server {
23 namespace kernel {
AddInputNameType(const std::string & name,TypeId type)24 ParamsInfo &ParamsInfo::AddInputNameType(const std::string &name, TypeId type) {
25 inputs_name_type_.push_back(std::make_pair(name, type));
26 inputs_names_.push_back(name);
27 return *this;
28 }
29
AddWorkspaceNameType(const std::string & name,TypeId type)30 ParamsInfo &ParamsInfo::AddWorkspaceNameType(const std::string &name, TypeId type) {
31 workspaces_name_type_.push_back(std::make_pair(name, type));
32 workspace_names_.push_back(name);
33 return *this;
34 }
35
AddOutputNameType(const std::string & name,TypeId type)36 ParamsInfo &ParamsInfo::AddOutputNameType(const std::string &name, TypeId type) {
37 outputs_name_type_.push_back(std::make_pair(name, type));
38 outputs_names_.push_back(name);
39 return *this;
40 }
41
inputs_num() const42 size_t ParamsInfo::inputs_num() const { return inputs_name_type_.size(); }
43
outputs_num() const44 size_t ParamsInfo::outputs_num() const { return outputs_name_type_.size(); }
45
inputs_name_type(size_t index) const46 const std::pair<std::string, TypeId> &ParamsInfo::inputs_name_type(size_t index) const {
47 if (index >= inputs_name_type_.size()) {
48 MS_LOG(EXCEPTION) << "Index " << index << " is out of bound of inputs_name_type_.";
49 }
50 return inputs_name_type_[index];
51 }
52
outputs_name_type(size_t index) const53 const std::pair<std::string, TypeId> &ParamsInfo::outputs_name_type(size_t index) const {
54 if (index >= outputs_name_type_.size()) {
55 MS_LOG(EXCEPTION) << "Index " << index << " is out of bound of outputs_name_type_.";
56 }
57 return outputs_name_type_[index];
58 }
59
inputs_names() const60 const std::vector<std::string> &ParamsInfo::inputs_names() const { return inputs_names_; }
61
workspace_names() const62 const std::vector<std::string> &ParamsInfo::workspace_names() const { return workspace_names_; }
63
outputs_names() const64 const std::vector<std::string> &ParamsInfo::outputs_names() const { return outputs_names_; }
65 } // namespace kernel
66 } // namespace server
67 } // namespace fl
68 } // namespace mindspore
69