Searched refs:referenced_tensors_set_ (Results 1 – 2 of 2) sorted by relevance
30 delete referenced_tensors_set_; in ~UniqueTensorReferences()37 if (referenced_tensors_set_ != nullptr) { in Add()41 if (!referenced_tensors_set_->insert(tensor_ref).second) { in Add()57 DCHECK(referenced_tensors_set_ == nullptr); in Add()58 referenced_tensors_set_ = new ReferencedTensorsSet; in Add()59 referenced_tensors_set_->reserve(kInVector); in Add()60 referenced_tensors_set_->insert(referenced_tensors_vector_.begin(), in Add()62 DCHECK_EQ(kInVector, referenced_tensors_set_->size()); in Add()73 if (referenced_tensors_set_ != nullptr) { in FreezeAndReturnReferences()75 out_vector->reserve(referenced_tensors_set_->size()); in FreezeAndReturnReferences()[all …]
36 UniqueTensorReferences() : frozen_(false), referenced_tensors_set_(nullptr) {} in UniqueTensorReferences()74 ReferencedTensorsSet* referenced_tensors_set_; variable