Home
last modified time | relevance | path

Searched refs:RefCountIsOne (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/lib/core/
Drefcount.h47 bool RefCountIsOne() const;
90 if (RefCountIsOne() || ref_.fetch_sub(1) == 1) { in Unref()
100 inline bool RefCounted::RefCountIsOne() const { in RefCountIsOne() function
Drefcount_test.cc64 ASSERT_TRUE(ref->RefCountIsOne()); in TEST_F()
71 ASSERT_FALSE(ref->RefCountIsOne()); in TEST_F()
/external/tensorflow/tensorflow/core/kernels/
Dtraining_op_helpers.h38 if (!tensor->RefCountIsOne()) { in PrepareToUpdateVariable()
Dbatch_util.cc97 bool can_move = element.RefCountIsOne(); in CopyElementToSlice()
Dresource_variable_ops.cc262 if (!variable->tensor()->RefCountIsOne() || in Compute()
/external/tensorflow/tensorflow/core/framework/
Dtensor.cc84 if (RefCountIsOne()) { in FillAllocationDescription()
662 bool Tensor::RefCountIsOne() const { in RefCountIsOne() function in tensorflow::Tensor
663 return buf_ != nullptr && buf_->RefCountIsOne() && in RefCountIsOne()
664 buf_->root_buffer()->RefCountIsOne() && buf_->OwnsMemory(); in RefCountIsOne()
Dtensor.h463 bool RefCountIsOne() const;
Dop_kernel.cc465 if (input.tensor == nullptr || input.is_ref() || !input->RefCountIsOne()) { in forward_input()
/external/tensorflow/tensorflow/core/lib/db/
Dsqlite.cc147 DCHECK((*db)->RefCountIsOne()); in Open()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dexecutor_test.cc369 while (!rendez_->RefCountIsOne()) in TEST_F()
/external/tensorflow/tensorflow/core/kernels/data/
Dtensor_queue_dataset_op.cc296 if (RefCountIsOne()) { in GetNext()
/external/tensorflow/tensorflow/c/
Dc_api.cc232 if (buf->RefCountIsOne() && buf->root_buffer()->RefCountIsOne() && in TF_TensorMaybeMove()