Searched refs:referenced_tensors_vector_ (Results 1 – 2 of 2) sorted by relevance
46 for (size_t i = 0; i < referenced_tensors_vector_.size(); ++i) { in Add()47 if (referenced_tensors_vector_[i].SharesBufferWith(tensor)) { in Add()52 referenced_tensors_vector_.push_back(TensorReference(tensor)); in Add()53 if (kInVector == referenced_tensors_vector_.size()) { in Add()60 referenced_tensors_set_->insert(referenced_tensors_vector_.begin(), in Add()61 referenced_tensors_vector_.end()); in Add()63 referenced_tensors_vector_.clear(); in Add()74 DCHECK(referenced_tensors_vector_.empty()); in FreezeAndReturnReferences()83 out_vector->reserve(referenced_tensors_vector_.size()); in FreezeAndReturnReferences()84 for (const auto& ref : referenced_tensors_vector_) { in FreezeAndReturnReferences()[all …]
66 TensorReferenceVector referenced_tensors_vector_; variable