Home
last modified time | relevance | path

Searched refs:cus_output_map_ (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dop_adapter.cc39 (*cus_output_map_)[prim->name()] = input_map; in GenerateCustomOpInputMap()
64 (*cus_output_map_)[prim->name()] = output_map; in GenerateCustomOpOutputMap()
75 if (cus_output_map_->find(prim->name()) == cus_output_map_->end()) { in GenerateCustomOpOutputMap()
76 (*cus_output_map_)[prim->name()] = output_map; in GenerateCustomOpOutputMap()
246 auto it = cus_output_map_->find(op->GetOpType()); in getCustomOutput()
247 if (it == cus_output_map_->end()) { in getCustomOutput()
289 if (cus_output_map_->find(op->GetOpType()) == cus_output_map_->end() || in UpdateSingleOutputDesc()
290 ((*cus_output_map_)[op->GetOpType()].empty())) { in UpdateSingleOutputDesc()
296 std::unordered_map<int, std::string> output_map = (*cus_output_map_)[op->GetOpType()]; in UpdateSingleOutputDesc()
310 if (cus_output_map_->find(cus_op->GetOpType()) == cus_output_map_->end()) { in GetCustomOpOutputSize()
[all …]
Dop_adapter.h52 cus_output_map_(cus_output_map), in OpAdapterImpl()
57 MS_EXCEPTION_IF_NULL(cus_output_map_); in OpAdapterImpl()
107 std::unordered_map<std::string, std::unordered_map<int, std::string>> *const cus_output_map_; variable
120 … &cus_output_map_, &extra_attr_, &name_counts_, this)) { in OpAdapter()
127 … &cus_output_map_, &extra_attr_, &name_counts_, this)) { in OpAdapter()
436 static std::unordered_map<std::string, std::unordered_map<int, std::string>> cus_output_map_; variable
461 std::unordered_map<std::string, std::unordered_map<int, std::string>> OpAdapter<T>::cus_output_map_;
Dop_adapter_map.cc25 …_map<std::string, std::unordered_map<int, std::string>> OpAdapter<ge::Operator>::cus_output_map_{};