Browse Source

update onnx format

tags/v1.2.0
y00500818 3 years ago
parent
commit
074e7d4f8c
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      ge/ir_build/atc_ir_common.h
  2. +1
    -1
      ge/offline/main.cc

+ 1
- 1
ge/ir_build/atc_ir_common.h View File

@@ -31,7 +31,7 @@
namespace ge { namespace ge {
static std::set<std::string> caffe_support_input_format = {"NCHW", "ND"}; 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> 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 = { static std::map<std::string, domiTensorFormat_t> input_format_str_to_geformat = {
{"ND", domi::DOMI_TENSOR_ND}, {"ND", domi::DOMI_TENSOR_ND},


+ 1
- 1
ge/offline/main.cc View File

@@ -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 kModelToJsonSupport = "only support 0(Caffe) 3(TensorFlow) 5(Onnx)";
const char *const kCaffeFormatSupport = "only support NCHW, ND in Caffe model"; 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 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 // limit available mem size 2G
const long kMinAvailableMem = 2097152; // 2 * 1024 * 1024 const long kMinAvailableMem = 2097152; // 2 * 1024 * 1024
} // namespace } // namespace


Loading…
Cancel
Save