Home
last modified time | relevance | path

Searched refs:referenced_tensors_vector_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dunique_tensor_references.cc46 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 …]
Dunique_tensor_references.h66 TensorReferenceVector referenced_tensors_vector_; variable