update onnx format

pull/1246/head
y00500818 4 years ago
parent 9f027029d5
commit 074e7d4f8c

@ -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},

@ -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

Loading…
Cancel
Save