|
@@ -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) { |
|
|
|
|
|
|
|
|
bool OpRegistrationTbe::Finalize(const OpRegistrationData ®_data, bool is_train) onst { |
|
|
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; |
|
|