Searched refs:graph_info_ (Results 1 – 2 of 2) sorted by relevance
42 graph_info_(std::move(graph_info)), in ArenaPlanner()65 allocs_.resize(graph_info_->num_tensors()); in ResetAllocations()72 TfLiteTensor& tensor = *graph_info_->tensor(i); in ResetAllocationsAfter()88 alloc_node_.assign(graph_info_->num_tensors(), kNodeNotAssigned); in PlanAllocations()89 dealloc_node_.assign(graph_info_->num_tensors(), kNodeNotAssigned); in PlanAllocations()92 std::vector<int> refcounts(graph_info_->num_tensors(), 0); in PlanAllocations()118 for (int tensor_index : graph_info_->outputs()) { in PlanAllocations()124 for (int tensor_index : graph_info_->variables()) { in PlanAllocations()137 for (int tensor_index : graph_info_->inputs()) { in PlanAllocations()147 for (size_t i = 0; i < graph_info_->num_execution_nodes(); ++i) { in PlanAllocations()[all …]
102 std::unique_ptr<GraphInfo> graph_info_; variable