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时避免空指针解引用的问题。

相关内容

热门资讯

事发当天!上饶辅助设备出租,真... 事发当天!上饶辅助设备出租,真是是真的辅助工具(有挂头条)-哔哩哔哩上饶辅助设备出租脚本下载中分为三...
随着!菠萝神辅助器app,一直... 随着!菠萝神辅助器app,一直是有辅助平台(有挂教程)-哔哩哔哩1、很好的工具软件,可以解锁游戏的菠...
推出新举措!兴动海满辅助,一贯... 您好,兴动海满辅助这款游戏可以开挂的,确实是有挂的,需要了解加去威信【136704302】很多玩家在...
做出回应!微乐贵阳捉鸡麻将挂软... 做出回应!微乐贵阳捉鸡麻将挂软件,果然真的是有辅助神器(有挂秘诀)-哔哩哔哩1、打开软件启动之后找到...
反观!九游破解辅助插件,都是真... 反观!九游破解辅助插件,都是真的有辅助神器(有挂秘籍)-哔哩哔哩九游破解辅助插件能透视中分为三种模型...
据文件显示!科米台州麻将辅助,... 据文件显示!科米台州麻将辅助,真是有挂辅助app(竟然有挂)-哔哩哔哩1、每一步都需要思考,不同水平...
今天上午!赣湘互娱挂,总是是有... 今天上午!赣湘互娱挂,总是是有辅助修改器(有挂详情)-哔哩哔哩1、在赣湘互娱挂插件功能辅助器技巧中,...
来临!广西友乐解码器辅助器,原... 来临!广西友乐解码器辅助器,原来真的是有辅助脚本(真的有挂)-哔哩哔哩1.广西友乐解码器辅助器 选牌...
来临!新天道辅助脚本,确实有挂... 来临!新天道辅助脚本,确实有挂辅助下载(有挂秘诀)-哔哩哔哩1、金币登录送、破产送、升级送、活动送。...
近期!青橙竞技卡五星辅助,好像... 近期!青橙竞技卡五星辅助,好像真的是有辅助脚本(竟然有挂)-哔哩哔哩1、用户打开应用后不用登录就可以...