|
|
|
@@ -0,0 +1,92 @@ |
|
|
|
/** |
|
|
|
* Copyright 2020 Huawei Technologies Co., Ltd |
|
|
|
* |
|
|
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
|
|
|
* you may not use this file except in compliance with the License. |
|
|
|
* You may obtain a copy of the License at |
|
|
|
* |
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0 |
|
|
|
* |
|
|
|
* Unless required by applicable law or agreed to in writing, software |
|
|
|
* distributed under the License is distributed on an "AS IS" BASIS, |
|
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
|
|
* See the License for the specific language governing permissions and |
|
|
|
* limitations under the License. |
|
|
|
*/ |
|
|
|
#ifndef MINDSPORE_CCSRC_MINDDATA_DATASET_INCLUDE_TYPEID_H_ |
|
|
|
#define MINDSPORE_CCSRC_MINDDATA_DATASET_INCLUDE_TYPEID_H_ |
|
|
|
|
|
|
|
#include "mindspore/core/ir/dtype/type_id.h" |
|
|
|
#include "minddata/dataset/core/data_type.h" |
|
|
|
|
|
|
|
namespace mindspore { |
|
|
|
namespace dataset { |
|
|
|
inline dataset::DataType MSTypeToDEType(TypeId data_type) { |
|
|
|
switch (data_type) { |
|
|
|
case kNumberTypeBool: |
|
|
|
return dataset::DataType(dataset::DataType::DE_BOOL); |
|
|
|
case kNumberTypeInt8: |
|
|
|
return dataset::DataType(dataset::DataType::DE_INT8); |
|
|
|
case kNumberTypeUInt8: |
|
|
|
return dataset::DataType(dataset::DataType::DE_UINT8); |
|
|
|
case kNumberTypeInt16: |
|
|
|
return dataset::DataType(dataset::DataType::DE_INT16); |
|
|
|
case kNumberTypeUInt16: |
|
|
|
return dataset::DataType(dataset::DataType::DE_UINT16); |
|
|
|
case kNumberTypeInt32: |
|
|
|
return dataset::DataType(dataset::DataType::DE_INT32); |
|
|
|
case kNumberTypeUInt32: |
|
|
|
return dataset::DataType(dataset::DataType::DE_UINT32); |
|
|
|
case kNumberTypeInt64: |
|
|
|
return dataset::DataType(dataset::DataType::DE_INT64); |
|
|
|
case kNumberTypeUInt64: |
|
|
|
return dataset::DataType(dataset::DataType::DE_UINT64); |
|
|
|
case kNumberTypeFloat16: |
|
|
|
return dataset::DataType(dataset::DataType::DE_FLOAT16); |
|
|
|
case kNumberTypeFloat32: |
|
|
|
return dataset::DataType(dataset::DataType::DE_FLOAT32); |
|
|
|
case kNumberTypeFloat64: |
|
|
|
return dataset::DataType(dataset::DataType::DE_FLOAT64); |
|
|
|
case kObjectTypeString: |
|
|
|
return dataset::DataType(dataset::DataType::DE_STRING); |
|
|
|
default: |
|
|
|
return dataset::DataType(dataset::DataType::DE_UNKNOWN); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
inline TypeId DETypeToMSType(dataset::DataType data_type) { |
|
|
|
switch (data_type.value()) { |
|
|
|
case dataset::DataType::DE_BOOL: |
|
|
|
return mindspore::TypeId::kNumberTypeBool; |
|
|
|
case dataset::DataType::DE_INT8: |
|
|
|
return mindspore::TypeId::kNumberTypeInt8; |
|
|
|
case dataset::DataType::DE_UINT8: |
|
|
|
return mindspore::TypeId::kNumberTypeUInt8; |
|
|
|
case dataset::DataType::DE_INT16: |
|
|
|
return mindspore::TypeId::kNumberTypeInt16; |
|
|
|
case dataset::DataType::DE_UINT16: |
|
|
|
return mindspore::TypeId::kNumberTypeUInt16; |
|
|
|
case dataset::DataType::DE_INT32: |
|
|
|
return mindspore::TypeId::kNumberTypeInt32; |
|
|
|
case dataset::DataType::DE_UINT32: |
|
|
|
return mindspore::TypeId::kNumberTypeUInt32; |
|
|
|
case dataset::DataType::DE_INT64: |
|
|
|
return mindspore::TypeId::kNumberTypeInt64; |
|
|
|
case dataset::DataType::DE_UINT64: |
|
|
|
return mindspore::TypeId::kNumberTypeUInt64; |
|
|
|
case dataset::DataType::DE_FLOAT16: |
|
|
|
return mindspore::TypeId::kNumberTypeFloat16; |
|
|
|
case dataset::DataType::DE_FLOAT32: |
|
|
|
return mindspore::TypeId::kNumberTypeFloat32; |
|
|
|
case dataset::DataType::DE_FLOAT64: |
|
|
|
return mindspore::TypeId::kNumberTypeFloat64; |
|
|
|
case dataset::DataType::DE_STRING: |
|
|
|
return mindspore::TypeId::kObjectTypeString; |
|
|
|
default: |
|
|
|
return kTypeUnknown; |
|
|
|
} |
|
|
|
} |
|
|
|
} // namespace dataset |
|
|
|
} // namespace mindspore |
|
|
|
|
|
|
|
#endif // MINDSPORE_CCSRC_MINDDATA_DATASET_INCLUDE_TYPEID_H_ |