Searched refs:set_ref_count (Results 1 – 7 of 7) sorted by relevance
112 void set_ref_count(int ref_count) override { in set_ref_count() function116 tensor->set_ref_count(ref_count); in set_ref_count()122 set_ref_count(this->init_ref_count_); in ResetRefCount()125 tensor->set_ref_count(this->init_ref_count_); in ResetRefCount()
35 tensor->set_ref_count(0); in Run()
154 virtual void set_ref_count(int ref_count) { in set_ref_count() function160 virtual void ResetRefCount() { set_ref_count(static_cast<int>(this->init_ref_count_)); } in ResetRefCount()
114 tensor_ptr->set_ref_count(this->ref_count()); in MallocTensorListData()
435 src_tensor->set_ref_count(1); in IsolateOutputTensor()
52 void set_ref_count(size_t ref_count) { ref_count_ = ref_count; } in set_ref_count() function
200 tensor->set_ref_count(refcount); in InitTensorsRef()