| @@ -1,92 +0,0 @@ | |||||
| /** | |||||
| * 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_ | |||||