提交 315ad6f7 编写于 作者: A Alan Kelly 提交者: TensorFlower Gardener

Fill refcounts with zeros so that arena memory may be properly shared.

PiperOrigin-RevId: 564693001
上级 9c775b06
......@@ -209,7 +209,7 @@ TfLiteStatus ArenaPlanner::PlanAllocations() {
std::max(graph_info_->num_execution_nodes(), (size_t)1), {});
// Keeps track of references to each tensor.
refcounts_.resize(num_tensors, 0);
refcounts_.assign(num_tensors, 0);
auto allocate = [this](int node, int tensor) -> TfLiteStatus {
if (alloc_node_[tensor] != kNodeNotAssigned) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册