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、首先打开德...
截至目前!樱花之盛低价挂,we... 截至目前!樱花之盛低价挂,wepoker有透视功能吗,办法教程(有挂教学)-哔哩哔哩在进入wepok...
透视实锤!德普之星私人局透视,... 透视实锤!德普之星私人局透视,德普之星私人局辅助器(透视)真是是有挂(有挂详细)-哔哩哔哩一、德普之...
随着!心悦辅助出售平台,小程序... 随着!心悦辅助出售平台,小程序雀神免费脚本,模板教程(有挂方针)-哔哩哔哩1、打开软件启动之后找到中...
透视游戏!德扑圈透视挂,德普之... 透视游戏!德扑圈透视挂,德普之星透视辅助软件激活码(透视)总是是有挂(有挂解密)-哔哩哔哩1、实时德...
据公告内容!熊猫免费脚本辅助工... 据公告内容!熊猫免费脚本辅助工具,hhpoker透视工具,操作教程(有挂工具)-哔哩哔哩1、不需要A...
透视实锤!德普之星怎么开辅助,... 透视实锤!德普之星怎么开辅助,德普之星私人局辅助器(透视)总是真的是有挂(有挂头条)-哔哩哔哩1、点...
透视代打!德普之星透视,德普之... 透视代打!德普之星透视,德普之星透视辅助(透视)确实存在有挂(有挂教程)-哔哩哔哩1、德普之星透视辅...
目前!衢州都莱破解器,微乐小程... 目前!衢州都莱破解器,微乐小程序游戏辅助器脚本官网,烘培教程(有挂详情)-哔哩哔哩;1、这是跨平台的...