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

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}};
if (is_train) {
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");
* you may not use this file except in compliance with the License.
@@ -24,7 +24,7 @@ class OpRegistrationTbe {
public:
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:
bool RegisterParser(const OpRegistrationData &reg_data) const;


Loading…
Cancel
Save