|
|
@@ -27,6 +27,7 @@ namespace kernel { |
|
|
static std::map<int32_t, int32_t> MS_PROTO_DATA_TYPE_MAP = { |
|
|
static std::map<int32_t, int32_t> MS_PROTO_DATA_TYPE_MAP = { |
|
|
{mindspore::TypeId::kTypeUnknown, mindspore::DataType::MS_UNKNOWN}, |
|
|
{mindspore::TypeId::kTypeUnknown, mindspore::DataType::MS_UNKNOWN}, |
|
|
{mindspore::TypeId::kNumberTypeBool, mindspore::DataType::MS_BOOL}, |
|
|
{mindspore::TypeId::kNumberTypeBool, mindspore::DataType::MS_BOOL}, |
|
|
|
|
|
{mindspore::TypeId::kNumberTypeInt, mindspore::DataType::MS_INT32}, |
|
|
{mindspore::TypeId::kNumberTypeInt8, mindspore::DataType::MS_INT8}, |
|
|
{mindspore::TypeId::kNumberTypeInt8, mindspore::DataType::MS_INT8}, |
|
|
{mindspore::TypeId::kNumberTypeInt16, mindspore::DataType::MS_INT16}, |
|
|
{mindspore::TypeId::kNumberTypeInt16, mindspore::DataType::MS_INT16}, |
|
|
{mindspore::TypeId::kNumberTypeInt32, mindspore::DataType::MS_INT32}, |
|
|
{mindspore::TypeId::kNumberTypeInt32, mindspore::DataType::MS_INT32}, |
|
|
@@ -34,8 +35,10 @@ static std::map<int32_t, int32_t> MS_PROTO_DATA_TYPE_MAP = { |
|
|
{mindspore::TypeId::kNumberTypeUInt, mindspore::DataType::MS_UINT32}, |
|
|
{mindspore::TypeId::kNumberTypeUInt, mindspore::DataType::MS_UINT32}, |
|
|
{mindspore::TypeId::kNumberTypeUInt8, mindspore::DataType::MS_UINT8}, |
|
|
{mindspore::TypeId::kNumberTypeUInt8, mindspore::DataType::MS_UINT8}, |
|
|
{mindspore::TypeId::kNumberTypeUInt16, mindspore::DataType::MS_UINT16}, |
|
|
{mindspore::TypeId::kNumberTypeUInt16, mindspore::DataType::MS_UINT16}, |
|
|
|
|
|
{mindspore::TypeId::kNumberTypeUInt32, mindspore::DataType::MS_UINT32}, |
|
|
{mindspore::TypeId::kNumberTypeUInt64, mindspore::DataType::MS_UINT64}, |
|
|
{mindspore::TypeId::kNumberTypeUInt64, mindspore::DataType::MS_UINT64}, |
|
|
{mindspore::TypeId::kNumberTypeFloat16, mindspore::DataType::MS_FLOAT16}, |
|
|
{mindspore::TypeId::kNumberTypeFloat16, mindspore::DataType::MS_FLOAT16}, |
|
|
|
|
|
{mindspore::TypeId::kNumberTypeFloat, mindspore::DataType::MS_FLOAT32}, |
|
|
{mindspore::TypeId::kNumberTypeFloat32, mindspore::DataType::MS_FLOAT32}, |
|
|
{mindspore::TypeId::kNumberTypeFloat32, mindspore::DataType::MS_FLOAT32}, |
|
|
{mindspore::TypeId::kNumberTypeFloat64, mindspore::DataType::MS_FLOAT64}, |
|
|
{mindspore::TypeId::kNumberTypeFloat64, mindspore::DataType::MS_FLOAT64}, |
|
|
}; |
|
|
}; |
|
|
|