AndroidNativeC++LibrarywithTFLite中AllocateTensors()函数的空指针解引用问题
创始人
2024-10-09 09:30:14
0次

在AllocateTensors()函数中添加条件判断,以确保输入tensor指针不为空。示例代码如下所示:

void Interpreter::AllocateTensors() {
  TensorAllocator* allocator = GetAllocator(0);
  for (int i = 0; i < tensors_.size(); ++i) {
    TfLiteTensor* t = &tensors_[i];
    if (t->allocation_type != kNoAllocation) {
      if (t->data.raw == nullptr || !allocator) {
        // 若数据指针为空或分配器未定义,则给出错误提示信息。
        LOG(ERROR) << "AllocateTensors: nullptr tensor or allocator";
        return;
      }
      if (t->bytes == 0 || TfLiteIntArrayProduct(t->dims) == 0) {
        // 零尺寸的tensor视为空tensor,不进行内存分配。
        t->data.raw = nullptr;
        continue;
      }
      size_t alignment = 8;
      t->bytes = (t->bytes + alignment - 1) & ~(alignment - 1);
      if (!allocator->Allocate(t)) {
        // 内存分配失败,给出错误提示信息。
        LOG(ERROR) << "AllocateTensors: failed to allocate tensor";
        return;
      }
    }
  }
  in_flight_execution_plan_->UpdateInputOutputTensorPointers(tensors_);
}

这样,就会在输入tensor为NULL时避免空指针解引用的问题。

相关内容

热门资讯

玩家必备教程!微乐自建房插件如... 玩家必备教程!微乐自建房插件如何安装(作弊器),微乐过炸确实是真的有透视挂方法微乐自建房插件如何安装...
揭秘关于!微乐小程序授权(作弊... 揭秘关于!微乐小程序授权(作弊器),微乐卡五星麻将真是有透视挂工具1)微乐小程序授权免费钻石:进一步...
今日科普!微乐自建房提升运势(... 今日科普!微乐自建房提升运势(外挂),微乐陕西三代竟然有透视挂攻略1、任何微乐自建房提升运势透视是真...
终于清楚!微乐智能辅助app(... 终于清楚!微乐智能辅助app(开挂),微乐三打一都是是真的有透视挂插件1、微乐智能辅助app辅助器安...
一分钟揭秘!微乐家乡小程序脚本... 一分钟揭秘!微乐家乡小程序脚本ios(辅助),微乐小程序都是真的有透视挂技巧一、微乐家乡小程序脚本i...
查到实测!微信微乐辅助器免费安... 查到实测!微信微乐辅助器免费安装(透视挂),微乐北京麻将好像真的是有透视挂攻略;1、金币登录送、破产...
一起来讨论!微信小程序微乐辅助... 一起来讨论!微信小程序微乐辅助器(外挂),微乐郑州麻将确实真的是有透视挂软件1)微信小程序微乐辅助器...
玩家必备攻略!微信小程序微乐跑... 玩家必备攻略!微信小程序微乐跑得快技巧(辅助),微乐麻将小程序原来是真的有透视挂软件在进入微信小程序...
查到实测辅助!微乐小程序自建房... 查到实测辅助!微乐小程序自建房透视免费(开挂),微乐划水麻将竟然是有透视挂攻略1、微乐小程序自建房透...
我来分享!广西微乐小程序辅助器... 我来分享!广西微乐小程序辅助器(透视器),微乐踢坑总是有透视挂脚本亲,关键说明,广西微乐小程序辅助器...