Browse Source

cleancode

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

+ 1
- 1
parser/common/pre_checker.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.


+ 1
- 1
parser/common/proto_file_parser.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.


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

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


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

@@ -24,7 +24,7 @@ class OpRegistrationTbe {
public: public:
static OpRegistrationTbe *Instance(); static OpRegistrationTbe *Instance();


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


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


Loading…
Cancel
Save