|
|
|
@ -162,18 +162,17 @@ void SetNodeAttr(const std::shared_ptr<AnfNode> &anf_node, mindspore::NodeDef *p
|
|
|
|
|
::google::protobuf::Map<::std::string, ::mindspore::AttrValue> *node_attr = proto->mutable_attrs();
|
|
|
|
|
for (const auto &attr_ptr : attrs_ptr) {
|
|
|
|
|
std::string attr_name = attr_ptr->name();
|
|
|
|
|
std::string real_name;
|
|
|
|
|
auto value = primitive->GetAttr(attr_name);
|
|
|
|
|
if (value != nullptr) {
|
|
|
|
|
if (attr_name == kQueueName || attr_name == kSharedName) {
|
|
|
|
|
real_name = kChannelName;
|
|
|
|
|
attr_name = kChannelName;
|
|
|
|
|
} else if (attr_name == kSeed) {
|
|
|
|
|
real_name = "seed";
|
|
|
|
|
attr_name = "seed";
|
|
|
|
|
} else if (attr_name == kSeed2) {
|
|
|
|
|
real_name = "seed2";
|
|
|
|
|
attr_name = "seed2";
|
|
|
|
|
}
|
|
|
|
|
std::string type = attr_ptr->type();
|
|
|
|
|
ParseAttrValue(type, real_name, value, node_attr);
|
|
|
|
|
ParseAttrValue(type, attr_name, value, node_attr);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
MS_LOG(INFO) << "Set node attr end!";
|
|
|
|
@ -182,7 +181,7 @@ void SetNodeAttr(const std::shared_ptr<AnfNode> &anf_node, mindspore::NodeDef *p
|
|
|
|
|
void SetNodeInputs(const std::shared_ptr<AnfNode> &anf_node, mindspore::NodeDef *proto) {
|
|
|
|
|
size_t input_num = AnfAlgo::GetInputTensorNum(anf_node);
|
|
|
|
|
if (input_num == 0) {
|
|
|
|
|
MS_LOG(INFO) << "Node [" << AnfAlgo::GetCNodeName(anf_node) << "] does not have input. ";
|
|
|
|
|
MS_LOG(INFO) << "Node [" << AnfAlgo::GetCNodeName(anf_node) << "] does not have input.";
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|