Searched refs:node_attrs (Results 1 – 4 of 4) sorted by relevance
43 auto node_attrs = node_primitive->attrs(); in IsCNodePrimitveEqual() local48 node_attrs.erase(attr); in IsCNodePrimitveEqual()51 if (main_attrs.size() != node_attrs.size()) { in IsCNodePrimitveEqual()56 [&node_attrs](const std::pair<std::string, ValuePtr> &item) -> bool { in IsCNodePrimitveEqual()60 auto iter = node_attrs.find(item.first); in IsCNodePrimitveEqual()61 if (iter == node_attrs.end()) { in IsCNodePrimitveEqual()
109 node_attrs = attrs.copy()110 node_attrs["Name"] = config_name111 specification = [config_type, node_attrs]
109 node_attrs = attrs.copy()110 node_attrs['Name'] = config_name111 specification = [config_type, node_attrs]
93 auto node_attrs = prim_node->attrs(); in SetAllReduceRecomputeFlag() local94 …if (node_attrs.find(RECOMPUTE_COMM_OP) != node_attrs.end() && !GetValue<bool>(node_attrs[RECOMPUTE… in SetAllReduceRecomputeFlag()