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

相关内容

热门资讯

微扑克有辅助挂!nzt德州辅助... 微扑克有辅助挂!nzt德州辅助软件,wpk俱乐部长期盈利打法,存在挂教程(有挂介绍);致您一封信;亲...
wepokeai代打的胜率!G... wepokeai代打的胜率!GG扑克辅助软件,wpk微扑克最新辅助,存在挂教程(有挂辅助挂);人气非...
德扑ai助手!aapoker俱... 1、德扑ai助手!aapoker俱乐部后台可以操作吗,wepoke辅助透视是真的,揭秘教程(有挂详情...
aapoker辅助工具存在!w... aapoker辅助工具存在!wopoker系统套路,aapoker可以设置牌局,普及教程(有挂规律)...
德州之星外挂!德扑之星,wpk... 德州之星外挂!德扑之星,wpk外挂存在,教你攻略(有挂透明)1、很好的工具软件,可以解锁游戏的德州之...
wepokeai机器人!aap... 您好,wepokeai机器人这款游戏可以开挂的,确实是有挂的,需要了解加微【136704302】很多...
智星德州菠萝外挂!wpk长期盈... 智星德州菠萝外挂!wpk长期盈利打法教学,微扑克辅助挂,德州教程(有挂介绍),您好,智星德州菠萝外挂...
wpk辅助挂!wepoke真的... wpk辅助挂!wepoke真的有挂嘛,微扑克辅助器是骗人,2025新版教程(有挂教学)1、在wepo...
aapoker辅助!德州软件后... aapoker辅助!德州软件后台控制,德扑ai智能系统,系统教程(有挂技巧)是一款可以让一直输的玩家...
德扑之星有作弊!云扑克德州有挂... 德扑之星有作弊!云扑克德州有挂吗,微扑克使用方法,新版2025教程(有挂详情)1、金币登录送、破产送...