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");
* 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");
* 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;
}

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}};
if (is_train) {
op_map[domi::TENSORFLOW] = &tensorflow_train_op_map;


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

@@ -24,7 +24,7 @@ class OpRegistrationTbe {
public:
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:
bool RegisterParser(const OpRegistrationData &reg_data) const;


Loading…
Cancel
Save