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

fix typo

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