提交 1153126d 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!5549 Add Tflite Model

Merge pull request !5549 from jianghui58/master
......@@ -89,7 +89,7 @@ int SpaceToBatchCPUKernel::ReSize() {
padded_in_shape_[2] = in_shape[2] + param->paddings_[2] + param->paddings_[3];
auto num_elements_padded = EnumElement(padded_in_shape_.data(), in_shape.size());
auto output_shape = out_tensors_[0]->shape();
auto pedding_h_size = output_shape[2] * output_shape[3] * sizeof(float);
auto pedding_h_size = padded_in_shape_[2] * output_shape[3] * sizeof(float);
pedding_h_data_ = reinterpret_cast<float *>(context_->allocator->Malloc(pedding_h_size));
if (pedding_h_data_ == nullptr) {
MS_LOG(ERROR) << "malloc pedding h data fail!";
......
......@@ -89,6 +89,7 @@ hiai_cv_labelDetectorModel_v2.tflite
#hiai_cv_labelDetectorModel_v3.tflite
hiai_cv_labelDetectorModel_v4.tflite
hiai_dress_detect.tflite
hiai_cv_saliencyDetectorModel.tflite
hiai_frozen_inference_graph.tflite
hiai_ghostnet.tflite
hiai_iMaxDN_RGB.tflite
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册