提交 931a005c 编写于 作者: Y yejianwu

fix typo

上级 24144647
...@@ -24,8 +24,6 @@ ...@@ -24,8 +24,6 @@
#include "mace/core/types.h" #include "mace/core/types.h"
#include "mace/public/mace.h" #include "mace/public/mace.h"
#ifdef MACE_ENABLE_OPENCL #ifdef MACE_ENABLE_OPENCL
#include "mace/core/runtime/opencl/opencl_runtime.h" #include "mace/core/runtime/opencl/opencl_runtime.h"
#endif // MACE_ENABLE_OPENCL #endif // MACE_ENABLE_OPENCL
......
...@@ -24,7 +24,6 @@ ...@@ -24,7 +24,6 @@
#include "mace/utils/logging.h" #include "mace/utils/logging.h"
namespace mace { namespace mace {
{% if model_type == 'source' %}
namespace {{tag}} { namespace {{tag}} {
{% for i in range(net.tensors|length) %} {% for i in range(net.tensors|length) %}
...@@ -133,7 +132,6 @@ void CreateMemoryArena(mace::MemoryArena *mem_arena) { ...@@ -133,7 +132,6 @@ void CreateMemoryArena(mace::MemoryArena *mem_arena) {
} // namespace } // namespace
{% endif %}
namespace {{tag}} { namespace {{tag}} {
......
...@@ -24,11 +24,12 @@ ...@@ -24,11 +24,12 @@
namespace mace { namespace mace {
namespace {{tag}} { namespace {{tag}} {
const unsigned char *LoadModelData(const char *model_data_file);
const unsigned char *LoadModelData(const std::string &model_data_file);
void UnloadModelData(const unsigned char *model_data); void UnloadModelData(const unsigned char *model_data);
NetDef CreateNet(); const std::shared_ptr<NetDef> CreateNet();
const std::string ModelName(); const std::string ModelName();
......
...@@ -89,8 +89,7 @@ def convert_to_source(net_def, model_checksum, weight_checksum, template_dir, ...@@ -89,8 +89,7 @@ def convert_to_source(net_def, model_checksum, weight_checksum, template_dir,
embed_model_data=embed_model_data, embed_model_data=embed_model_data,
winograd_conv=winograd_conv, winograd_conv=winograd_conv,
checksum=checksum, checksum=checksum,
build_time=build_time, build_time=build_time)
model_type=model_load_type)
with open(output, "wb") as f: with open(output, "wb") as f:
f.write(source) f.write(source)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册