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

相关内容

热门资讯

透视肯定!aapoker真的假... 透视肯定!aapoker真的假的(透视)透视方法(一贯真的有挂)aapoker真的假的是一种具有地方...
透视辅助!aapoker透视怎... 透视辅助!aapoker透视怎么用(透视)透视脚本下载(确实真的是有挂)1、进入到aapoker透视...
透视玄学!aapoker如何设... 透视玄学!aapoker如何设置胜率(透视)透视脚本(确实是真的有挂);1.aapoker如何设置胜...
透视ai代打!aapoker透... 透视ai代打!aapoker透视脚本(透视)ai插件(一贯是有挂);1、任何aapoker透视脚本a...
透视神器!aapoker怎么开... 透视神器!aapoker怎么开辅助器(透视)发牌逻辑(原来有挂)在进入aapoker怎么开辅助器辅助...
透视模拟器!aapoker透视... 透视模拟器!aapoker透视脚本下载(透视)透视脚本(切实存在有挂)进入游戏-大厅左侧-新手福利-...
透视工具!aapoker辅助软... 透视工具!aapoker辅助软件合法吗(透视)透视脚本入口(总是有挂)1、打开软件启动之后找到中间准...
透视好友房!aapoker怎么... 透视好友房!aapoker怎么提高中牌率(透视)透视插件(本来是真的有挂)1、下载好aapoker怎...
透视代打!aa poker透视... 透视代打!aa poker透视软件(透视)透视脚本(其实真的是有挂)1、操作简单,无需注册,只需要使...
透视辅助!aapoker真的假... 透视辅助!aapoker真的假的(透视)辅助插件工具(竟然真的有挂);1)aapoker真的假的辅助...