Rename proto_utils to arg_helper.

mace/core/proto_utils.cc → mace/core/arg_helper.cc

@@ -2,7 +2,7 @@
 // Copyright (c) 2017 XiaoMi All rights reserved.
 //
 
-#include "mace/core/proto_utils.h"
+#include "mace/core/arg_helper.h"
 
 namespace mace {
 

mace/core/proto_utils.h → mace/core/arg_helper.h

@@ -2,8 +2,8 @@
 // Copyright (c) 2017 XiaoMi All rights reserved.
 //
 
-#ifndef MACE_CORE_PROTO_UTILS_H_
-#define MACE_CORE_PROTO_UTILS_H_
+#ifndef MACE_CORE_ARG_HELPER_H_
+#define MACE_CORE_ARG_HELPER_H_
 
 #include <map>
 
@@ -68,4 +68,4 @@ class ArgumentHelper {
 
 }  // namespace mace
 
-#endif  // MACE_CORE_PROTO_UTILS_H_
+#endif  // MACE_CORE_ARG_HELPER_H_

mace/core/operator.h

@@ -6,7 +6,7 @@
 #define MACE_CORE_OPERATOR_H
 
 #include "mace/core/common.h"
-#include "mace/core/proto_utils.h"
+#include "mace/core/arg_helper.h"
 #include "mace/core/registry.h"
 #include "mace/core/tensor.h"
 #include "mace/core/workspace.h"

mace/core/workspace.cc

@@ -4,7 +4,7 @@
 
 #include "mace/core/workspace.h"
 #include "mace/core/serializer.h"
-#include "mace/core/proto_utils.h"
+#include "mace/core/arg_helper.h"
 
 namespace mace {
 
@@ -69,7 +69,7 @@ void Workspace::LoadModelTensor(const NetDef &net_def, DeviceType type) {
 }
 
 void Workspace::CreateImageOutputTensor(const NetDef &net_def) {
-  if (net_def.has_mem_arena() || net_def.mem_arena().mem_block_size() == 0) {
+  if (!net_def.has_mem_arena() || net_def.mem_arena().mem_block_size() == 0) {
     return;
   }
   std::map<std::string, std::shared_ptr<Tensor>> mem_tensor_map;

mace/proto/BUILD

@@ -10,16 +10,6 @@ licenses(["notice"])  # Apache 2.0
 
 load("@com_google_protobuf//:protobuf.bzl", "py_proto_library")
 
-proto_library(
-    name = "proto",
-    srcs = ["mace.proto"],
-)
-
-cc_proto_library(
-    name = "cc_proto",
-    deps = [":proto"],
-)
-
 proto_library(
     name = "stats",
     srcs = ["stats.proto"],