@@ -1,5 +1,5 @@ | |||||
/** | /** | ||||
* Copyright 2020 Huawei Technologies Co., Ltd | |||||
* Copyright (c) Huawei Technologies Co., Ltd. 2022. All rights reserved. | |||||
* | * | ||||
* Licensed under the Apache License, Version 2.0 (the "License"); | * Licensed under the Apache License, Version 2.0 (the "License"); | ||||
* you may not use this file except in compliance with the License. | * you may not use this file except in compliance with the License. | ||||
@@ -1,5 +1,5 @@ | |||||
/** | /** | ||||
* Copyright 2020 Huawei Technologies Co., Ltd | |||||
* Copyright (c) Huawei Technologies Co., Ltd. 2022. All rights reserved. | |||||
* Licensed under the Apache License, Version 2.0 (the "License"); | * Licensed under the Apache License, Version 2.0 (the "License"); | ||||
* you may not use this file except in compliance with the License. | * you may not use this file except in compliance with the License. | ||||
@@ -42,7 +42,7 @@ FMK_FUNC_HOST_VISIBILITY OpRegistrationTbe *OpRegistrationTbe::Instance() { | |||||
return &instance; | return &instance; | ||||
} | } | ||||
bool OpRegistrationTbe::Finalize(const OpRegistrationData ®_data, bool is_train) const { | |||||
bool OpRegistrationTbe::Finalize(const OpRegistrationData ®_data, bool is_train) { | |||||
static std::map<domi::FrameworkType, std::map<std::string, std::string> *> op_map = {{domi::CAFFE, &caffe_op_map}}; | static std::map<domi::FrameworkType, std::map<std::string, std::string> *> op_map = {{domi::CAFFE, &caffe_op_map}}; | ||||
if (is_train) { | if (is_train) { | ||||
op_map[domi::TENSORFLOW] = &tensorflow_train_op_map; | op_map[domi::TENSORFLOW] = &tensorflow_train_op_map; | ||||
@@ -24,7 +24,7 @@ class OpRegistrationTbe { | |||||
public: | public: | ||||
static OpRegistrationTbe *Instance(); | static OpRegistrationTbe *Instance(); | ||||
bool Finalize(const OpRegistrationData ®_data, bool is_train = false) const; | |||||
bool Finalize(const OpRegistrationData ®_data, bool is_train = false); | |||||
private: | private: | ||||
bool RegisterParser(const OpRegistrationData ®_data) const; | bool RegisterParser(const OpRegistrationData ®_data) const; | ||||