Browse Source

add new available gts model

tags/v1.1.0
zengxianglong 5 years ago
parent
commit
dded908fae
2 changed files with 3 additions and 5 deletions
  1. +2
    -5
      mindspore/lite/src/runtime/kernel/arm/fp32/cast.cc
  2. +1
    -0
      mindspore/lite/test/models_tflite_awaretraining.cfg

+ 2
- 5
mindspore/lite/src/runtime/kernel/arm/fp32/cast.cc View File

@@ -71,7 +71,8 @@ int CastCPUKernel::DoCast(int thread_id) {
auto input_data_type = input->data_type();
auto output_data_type = output->data_type();
if (output_data_type != kNumberTypeFloat32) {
if (input_data_type == kNumberTypeFloat32 && output_data_type == kNumberTypeInt32) {
if (input_data_type == kNumberTypeFloat32 &&
(output_data_type == kNumberTypeInt32 || output_data_type == kNumberTypeInt64)) {
Float32ToInt32(reinterpret_cast<float *>(input->data_c()) + offset,
reinterpret_cast<int32_t *>(output_data) + offset, data_num);
} else if (input_data_type == kNumberTypeFloat32 && output_data_type == kNumberTypeFloat16) {
@@ -81,10 +82,6 @@ int CastCPUKernel::DoCast(int thread_id) {
(output_data_type == kNumberTypeInt32 || output_data_type == kNumberTypeInt64)) {
memcpy(reinterpret_cast<int32_t *>(output_data) + offset, reinterpret_cast<int32_t *>(input->data_c()) + offset,
data_num * sizeof(int32_t));
} else if (input_data_type == kNumberTypeFloat32 &&
(output_data_type == kNumberTypeInt32 || output_data_type == kNumberTypeInt64)) {
memcpy(reinterpret_cast<float *>(output_data) + offset, reinterpret_cast<float *>(input->data_c()) + offset,
data_num * sizeof(float));
} else {
MS_LOG(ERROR) << "Unsupported datatype from " << input_data_type << " to " << output_data_type;
return RET_ERROR;


+ 1
- 0
mindspore/lite/test/models_tflite_awaretraining.cfg View File

@@ -36,3 +36,4 @@ vision_classifier_fungi_mobile_V1_1_default_1.tflite
detect.tflite
ssd_mobilenet_v1_1_default_1.tflite
object_detection_mobile_object_localizer_v1_1_default_1.tflite
gts_detect_0730_quant_frozen.tflite

Loading…
Cancel
Save