Searched refs:set_original_ref_count (Results 1 – 6 of 6) sorted by relevance
50 …void set_original_ref_count(size_t original_ref_count) { original_ref_count_ = original_ref_count;… in set_original_ref_count() function
296 device_sync_->set_original_ref_count(SIZE_MAX);
98 new_device_tensor->set_original_ref_count(device_tensor->original_ref_count()); in UpdateOutputDeviceAddress()
173 device_tensor->set_original_ref_count(SIZE_MAX); in UpdateRefCount()
274 device_address->set_original_ref_count(SIZE_MAX); in SetSummaryNodesRefCount()286 device_address->set_original_ref_count(SIZE_MAX); in UpdateRefCountForGraphOutput()
255 new_device_tensor->set_original_ref_count(device_tensor->original_ref_count()); in UpdateOutputDeviceAddress()