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