!667 minor fixes for latest ascend package

From: @nicholas_yhr
Reviewed-by: @liujunzhu,@youui
Signed-off-by: @youui
pull/667/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit e4da33cca6

@ -37,6 +37,7 @@ enum FrameworkType {
MINDSPORE = 1,
TENSORFLOW = 3,
ANDROID_NN,
ONNX,
FRAMEWORK_RESERVED,
};

@ -118,7 +118,8 @@ const std::map<std::string, Format> kDataFormatMap = {
{"NCDHW", FORMAT_NCDHW},
{"ND", FORMAT_ND}};
const std::map<std::string, Format> kStringToFormatMap = {
const std::map<std::string, Format> kStringToFormatMap =
{
{"NCHW", FORMAT_NCHW},
{"NHWC", FORMAT_NHWC},
{"ND", FORMAT_ND},
@ -160,7 +161,11 @@ const std::map<std::string, Format> kStringToFormatMap = {
{"FRACTAL_Z_G", FORMAT_FRACTAL_Z_G},
{"FORMAT_RESERVED", FORMAT_RESERVED},
{"ALL", FORMAT_ALL},
{"NULL", FORMAT_NULL}};
{"NULL", FORMAT_NULL},
// add for json input
{"RESERVED", FORMAT_RESERVED},
{"UNDEFINED", FORMAT_RESERVED},
};
const std::map<DataType, std::string> kDataTypeToStringMap = {
{DT_UNDEFINED, "DT_UNDEFINED"}, // Used to indicate a DataType field has not been set.

Loading…
Cancel
Save