Searched refs:op_key (Results 1 – 3 of 3) sorted by relevance
43 SplitOpKey op_key(operator_type, type_id, is_depth_wise); in RegisterOperatorInfo() local44 if (operator_info_map_.find(op_key) != operator_info_map_.end()) { in RegisterOperatorInfo()45 MS_LOG(ERROR) << " Operator already exist " << op_key.ToString(); in RegisterOperatorInfo()48 …this->operator_info_map_.insert(std::pair<SplitOpKey, OperatorInfoCreatorFunc>(op_key, creator_fun… in RegisterOperatorInfo()51 OperatorInfoCreatorFunc OperatorInfoFactory::FindOperatorInfo(const SplitOpKey &op_key) { in FindOperatorInfo() argument52 auto iterator = this->operator_info_map_.find(op_key); in FindOperatorInfo()56 MS_LOG(ERROR) << "operator_info do not register" << op_key.ToString(); in FindOperatorInfo()
84 SplitOpKey op_key = SplitOpKey(split_schema_id, split_type_id, is_depth_wise_); in CreateParallelOperator() local85 auto op_create_func = OperatorInfoFactory::GeInstance()->FindOperatorInfo(op_key); in CreateParallelOperator()
109 for (auto op_key : optional_key_) { in InitAndLoad() local110 if (schema.find(op_key.first) != schema.end()) { in InitAndLoad()111 optional_key_[op_key.first] = true; in InitAndLoad()