From 074e7d4f8cf84800d820f8383b052c2ccf08dac2 Mon Sep 17 00:00:00 2001
From: y00500818 <yangyongqiang7@hisilicon.com>
Date: Fri, 12 Mar 2021 09:36:41 +0800
Subject: [PATCH] update onnx format

---
 ge/ir_build/atc_ir_common.h | 2 +-
 ge/offline/main.cc          | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/ge/ir_build/atc_ir_common.h b/ge/ir_build/atc_ir_common.h
index e8637cb9..6ff40547 100644
--- a/ge/ir_build/atc_ir_common.h
+++ b/ge/ir_build/atc_ir_common.h
@@ -31,7 +31,7 @@
 namespace ge {
 static std::set<std::string> caffe_support_input_format = {"NCHW", "ND"};
 static std::set<std::string> tf_support_input_format = {"NCHW", "NHWC", "ND", "NCDHW", "NDHWC"};
-static std::set<std::string> onnx_support_input_format = {"NCHW", "ND"};
+static std::set<std::string> onnx_support_input_format = {"NCHW", "ND", "NCDHW"};
 
 static std::map<std::string, domiTensorFormat_t> input_format_str_to_geformat = {
     {"ND", domi::DOMI_TENSOR_ND},
diff --git a/ge/offline/main.cc b/ge/offline/main.cc
index e2741e20..69ee29de 100755
--- a/ge/offline/main.cc
+++ b/ge/offline/main.cc
@@ -70,7 +70,7 @@ const char *const kModeSupport = "only support 0(model to framework model), "
 const char *const kModelToJsonSupport = "only support 0(Caffe) 3(TensorFlow) 5(Onnx)";
 const char *const kCaffeFormatSupport = "only support NCHW, ND in Caffe model";
 const char *const kTFFormatSupport = "only support NCHW, NHWC, ND, NCDHW, NDHWC in TF model";
-const char *const kONNXFormatSupport = "only support NCHW, ND in ONNX model";
+const char *const kONNXFormatSupport = "only support NCHW, ND, NCDHW in ONNX model";
 // limit available mem size 2G
 const long kMinAvailableMem = 2097152;  // 2 * 1024 * 1024
 }  // namespace