Searched refs:attr_map (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_custom_parser.cc | 83 auto attr_map = flexbuffers::GetRoot(custom_attr).AsMap(); in DetectPostProcess() local
84 …prim->set_scale({attr_map["h_scale"].AsFloat(), attr_map["w_scale"].AsFloat(), attr_map["x_scale"]… in DetectPostProcess()
85 attr_map["y_scale"].AsFloat()}); in DetectPostProcess()
86 prim->set_nms_iou_threshold(attr_map["nms_iou_threshold"].AsFloat()); in DetectPostProcess()
87 prim->set_nms_score_threshold(attr_map["nms_score_threshold"].AsFloat()); in DetectPostProcess()
88 prim->set_max_detections(attr_map["max_detections"].AsInt64()); in DetectPostProcess()
89 if (attr_map["detections_per_class"].IsNull()) { in DetectPostProcess()
92 prim->set_detections_per_class(attr_map["detections_per_class"].AsInt64()); in DetectPostProcess()
94 prim->set_max_classes_per_detection(attr_map["max_classes_per_detection"].AsInt64()); in DetectPostProcess()
95 prim->set_num_classes(attr_map["num_classes"].AsInt64()); in DetectPostProcess()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | resource.cc | 219 static BuiltInTypeMap attr_map = { in GetAttrMap() local
244 return attr_map; in GetAttrMap()
292 const BuiltInTypeMap &attr_map = GetAttrMap(); in IsTypeInBuiltInMap() local
293 iter = attr_map.find(static_cast<int64_t>(type_id)); in IsTypeInBuiltInMap()
294 if (iter == attr_map.end()) { in IsTypeInBuiltInMap()
309 const BuiltInTypeMap &attr_map = GetAttrMap(); in GetAttrPtr() local
310 return GetMethodOrAttr(name, type_id, attr_map); in GetAttrPtr()
|
| /third_party/boost/libs/graph/example/ |
| D | reachable-loop-head.cpp | 83 std::map< std::string, std::string >& attr_map = eattr_map[*ei]; in main() local
85 = attr_map.begin(); in main()
86 eai != attr_map.end(); ++eai) in main()
89 if (next(eai) != attr_map.end()) in main()
|
| D | loops_dfs.cpp | 184 std::map< std::string, std::string >& attr_map = eattr_map[*ei]; in main() local
186 = attr_map.begin(); in main()
187 eai != attr_map.end(); ++eai) in main()
190 if (next(eai) != attr_map.end()) in main()
|
| D | edge-connectivity.cpp | 176 property_map< GraphvizGraph, vertex_attribute_t >::type attr_map in main() local
182 std::cout << "(" << attr_map[source(*i, g)]["label"] << "," in main()
183 << attr_map[target(*i, g)]["label"] << ") "; in main()
|
| /third_party/mindspore/mindspore/ccsrc/debug/debugger/offline_debug/ |
| D | dbg_services.cc | 78 auto attr_map = node.second; in AddWatchpoint() local
80 bool is_output = std::get<bool>(attr_map["is_output"]); in AddWatchpoint()
83 std::vector<std::string> rank_id_str = std::get<std::vector<std::string>>(attr_map["rank_id"]); in AddWatchpoint()
93 …std::vector<std::string> root_graph_id_str = std::get<std::vector<std::string>>(attr_map["root_gra… in AddWatchpoint()
119 auto attr_map = node.second; in AddWatchpoint() local
120 return std::make_tuple(node.first, std::get<bool>(attr_map["is_output"])); in AddWatchpoint()
125 auto attr_map = node.second; in AddWatchpoint() local
126 … std::vector<std::string> rank_id_str = std::get<std::vector<std::string>>(attr_map["rank_id"]); in AddWatchpoint()
138 auto attr_map = node.second; in AddWatchpoint() local
139 …std::vector<std::string> root_graph_id_str = std::get<std::vector<std::string>>(attr_map["root_gra… in AddWatchpoint()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/ |
| D | op_adapter.h | 36 const std::unordered_map<std::string, AttrDesc> &attr_map, in OpAdapterImpl() argument
48 attr_map_(attr_map), in OpAdapterImpl()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | session_basic.cc | 1208 const auto &attr_map = prim->attrs(); in GetSingleOpGraphInfo() local
1209 (void)std::for_each(attr_map.begin(), attr_map.end(), [&](const auto &element) { in GetSingleOpGraphInfo()
|
| /third_party/mindspore/mindspore/lite/src/ops/ |
| D | ops_utils.cc | 1050 auto attr_map = ms_primc->get_attr(); in CustomPrimitiveCreator() local
1051 for (const auto &attr_item : attr_map) { in CustomPrimitiveCreator()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_execute.cc | 351 const auto &attr_map = op_prim->attrs(); in GetSingleOpGraphInfo() local
352 (void)std::for_each(attr_map.begin(), attr_map.end(), in GetSingleOpGraphInfo()
|