diff --git a/docs/development/memory_layout.rst b/docs/development/memory_layout.rst index 186da713ad83db1839e9087d6c40ceb69ca6b38c..81e4db432715596c61c737b3341422e4416d6320 100644 --- a/docs/development/memory_layout.rst +++ b/docs/development/memory_layout.rst @@ -50,11 +50,11 @@ The Input/Output Tensor is stored in NHWC format: * - Height-Major Input/Output - NHWC - [W * C, N * (H+3)/4] - - Winograd Convolution format + - WinogradTransform and MatMul output format * - Width-Major Input/Output - NHWC - [(W+3)/4 * C, N * H] - - Winograd Convolution format + - Unused now Each Pixel of **Image** contains 4 elements. The below table list the coordination relation between **Image** and **Buffer**. diff --git a/mace/core/runtime/opencl/opencl_runtime.cc b/mace/core/runtime/opencl/opencl_runtime.cc index f901973fce1cfb24754b662e42048dde4f373d44..305337034d98622db928d9133c4cc69597900ffe 100644 --- a/mace/core/runtime/opencl/opencl_runtime.cc +++ b/mace/core/runtime/opencl/opencl_runtime.cc @@ -321,7 +321,8 @@ OpenCLRuntime::OpenCLRuntime(): std::stringstream ss; ss << default_platform.getInfo() << ", " << default_platform.getInfo() << ", " - << default_platform.getInfo(); + << default_platform.getInfo() << ", " + << MaceVersion(); platform_info_ = ss.str(); VLOG(1) << "Using platform: " << platform_info_; diff --git a/mace/python/tools/model_header.jinja2 b/mace/python/tools/model_header.jinja2 index 28b1b54563943509e3b4d2ec2ef0c8a90bc87e9c..c90c4f08b266e209fecf37fd5969efb73a8f0e05 100644 --- a/mace/python/tools/model_header.jinja2 +++ b/mace/python/tools/model_header.jinja2 @@ -27,8 +27,6 @@ namespace {{tag}} { const unsigned char *LoadModelData(const std::string &model_data_file); -void UnloadModelData(const unsigned char *model_data); - const std::shared_ptr CreateNet(); const std::string ModelName();