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

相关内容

热门资讯

微扑克辅助器ios!微扑克ai... 1、微扑克辅助器ios!微扑克ai软件,微扑克有辅助插件(原来真的有挂);该软件可以轻松地帮助玩家将...
德州ai辅助有用!德州ai辅助... 德州ai辅助有用!德州ai辅助nzp,德州真是是有挂(详细有外 挂教程)1)德州ai辅助有用辅助挂:...
智星德州菠萝作弊器!wpk透视... 智星德州菠萝作弊器!wpk透视辅助器,德扑之星隐藏功能在哪(详细透视挂教程);(需添加指定薇7574...
wpk透明挂!WPk都是真的有... wpk透明挂!WPk都是真的有挂,wPk果然真的有挂,2025新版总结(有挂解密)wpk软件透明挂微...
wepoke黑科技!wepok... 1、wepoke黑科技!wepokeai代打辅助(辅助挂)wEPoke(一贯真的有挂)2、进入游戏-...
德扑之星有猫腻!德扑ai机器人... 1、德扑之星有猫腻!德扑ai机器人哪里有卖,德州扑克一贯是真的有挂(详细作弊教程)2、进入游戏-大厅...
微扑克全自动机器人!微扑克有规... 微扑克全自动机器人!微扑克有规律(辅助挂)一直真的是有挂1、完成微扑克透视辅助安装,帮助玩家取得所有...
WPK透视辅助!wpk德州辅助... WPK透视辅助!wpk德州辅助,wpK本来真的有挂,详细教程,(有挂揭秘)1、金币登录送、破产送、升...
wepower辅助器!德州竞技... wepower辅助器!德州竞技联盟辅助,轰趴大菠萝十三水辅助(详细辅助挂教程);玩家必备必赢加哟《1...
wepoke计算辅助!wepo... 1、wepoke计算辅助!wepower插件,wopoker轻量版外 挂,软件教程(有挂攻略);详细...