|
|
@@ -85,6 +85,9 @@ int CastCPUKernel::DoCast(int thread_id) { |
|
|
} else if (input_data_type == kNumberTypeInt32 && output_data_type == kNumberTypeInt64) { |
|
|
} else if (input_data_type == kNumberTypeInt32 && output_data_type == kNumberTypeInt64) { |
|
|
Int32ToInt64(reinterpret_cast<int32_t *>(input->data_c()) + offset, |
|
|
Int32ToInt64(reinterpret_cast<int32_t *>(input->data_c()) + offset, |
|
|
reinterpret_cast<int64_t *>(output_data) + offset, data_num); |
|
|
reinterpret_cast<int64_t *>(output_data) + offset, data_num); |
|
|
|
|
|
} else if (input_data_type == kNumberTypeFloat32 && output_data_type == kNumberTypeInt16) { |
|
|
|
|
|
Float32ToInt16(reinterpret_cast<float *>(input->data_c()) + offset, |
|
|
|
|
|
reinterpret_cast<int16_t *>(output_data) + offset, data_num); |
|
|
} else { |
|
|
} else { |
|
|
MS_LOG(ERROR) << "Unsupported datatype from " << input_data_type << " to " << output_data_type; |
|
|
MS_LOG(ERROR) << "Unsupported datatype from " << input_data_type << " to " << output_data_type; |
|
|
return RET_ERROR; |
|
|
return RET_ERROR; |
|
|
|