Searched refs:alloc_node_ (Results 1 – 2 of 2) sorted by relevance
88 alloc_node_.assign(graph_info_->num_tensors(), kNodeNotAssigned); in PlanAllocations()95 if (alloc_node_[tensor] != kNodeNotAssigned) { in PlanAllocations()100 alloc_node_[tensor] = node; in PlanAllocations()105 if (alloc_node_[tensor] == kNodeNotAssigned) { in PlanAllocations()194 alloc_node_.resize(graph_info_->num_tensors(), kNodeNotAssigned); in ExecuteAllocations()205 alloc_node_[tensor_index] = i; in ExecuteAllocations()268 if (this->alloc_node_[idx1] == 0 && in CreateTensorAllocationVector()270 if (this->alloc_node_[idx2] == 0 && in CreateTensorAllocationVector()276 if (this->alloc_node_[idx2] == 0 && in CreateTensorAllocationVector()288 return this->alloc_node_[idx1] < this->alloc_node_[idx2]; in CreateTensorAllocationVector()[all …]
109 std::vector<int32_t> alloc_node_; variable