Searched refs:graph_info_ (Results 1 – 2 of 2) sorted by relevance
34 graph_info_(std::move(graph_info)), in ArenaPlanner()57 allocs_.resize(graph_info_->num_tensors()); in ResetAllocations()71 std::vector<int> refcounts(graph_info_->num_tensors(), 0); in PlanAllocations()74 std::vector<int> allocated(graph_info_->num_tensors(), false); in PlanAllocations()75 std::vector<int> deallocated(graph_info_->num_tensors(), false); in PlanAllocations()102 alloc_queue_.reserve(2 * graph_info_->num_tensors()); in PlanAllocations()107 for (int tensor_index : graph_info_->outputs()) { in PlanAllocations()113 for (int tensor_index : graph_info_->variables()) { in PlanAllocations()119 for (int tensor_index : graph_info_->inputs()) { in PlanAllocations()129 for (int tensor_index : graph_info_->variables()) { in PlanAllocations()[all …]
95 std::unique_ptr<GraphInfo> graph_info_; variable