Browse Source

cleancode

pull/636/head
songmingyang 2 years ago
parent
commit
a00a617db5
2 changed files with 4 additions and 4 deletions
  1. +2
    -2
      parser/common/register_tbe.cc
  2. +2
    -2
      parser/common/register_tbe.h

+ 2
- 2
parser/common/register_tbe.cc View File

@@ -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 &reg_data, bool is_train) {
bool OpRegistrationTbe::Finalize(const OpRegistrationData &reg_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;


+ 2
- 2
parser/common/register_tbe.h View File

@@ -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.
@@ -24,7 +24,7 @@ class OpRegistrationTbe {
public: public:
static OpRegistrationTbe *Instance(); static OpRegistrationTbe *Instance();


bool Finalize(const OpRegistrationData &reg_data, bool is_train = false);
bool Finalize(const OpRegistrationData &reg_data, bool is_train = false) const;


private: private:
bool RegisterParser(const OpRegistrationData &reg_data) const; bool RegisterParser(const OpRegistrationData &reg_data) const;


Loading…
Cancel
Save