Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/lib/core/
Drefcount.h49 bool RefCountIsOne() const;
101 if (RefCountIsOne() || ref_.fetch_sub(1) == 1) { in Unref()
111 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/compiler/jit/
Dxla_tensor.cc30 /*static*/ bool XlaTensor::RefCountIsOne(const Tensor& tensor) { in RefCountIsOne() function in tensorflow::XlaTensor
31 return tensor.RefCountIsOne(); in RefCountIsOne()
Dxla_tensor.h42 static bool RefCountIsOne(const Tensor& tensor);
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_compilation_cache.cc63 while (!entry.second->RefCountIsOne()) { in ~XRTCompilationCache()
102 if (entry->RefCountIsOne()) { in DiscardEntryRefLocked()
Dxrt_state.cc494 if (!buffer.second->RefCountIsOne()) return false; in IsExclusiveOwner()
/external/tensorflow/tensorflow/core/kernels/
Dtraining_op_helpers.h39 if (var->tensor()->RefCountIsOne()) { in EnsureSparseVariableAccess()
201 if (copy_on_read_mode || !tensor->RefCountIsOne()) { in PrepareToUpdateVariable()
Dresource_variable_ops.cc441 if (!variable->tensor()->RefCountIsOne() || in Compute()
/external/tensorflow/tensorflow/core/util/
Dbatch_util.cc127 bool can_move = element.RefCountIsOne(); in CopyElementToSlice()
172 bool can_move = parent->RefCountIsOne(); in MaybeMoveSliceToElement()
/external/tensorflow/tensorflow/core/framework/
Dtensor.cc89 if (RefCountIsOne()) { in FillAllocationDescription()
682 bool Tensor::RefCountIsOne() const { in RefCountIsOne() function in tensorflow::Tensor
683 return buf_ != nullptr && buf_->RefCountIsOne() && in RefCountIsOne()
684 buf_->root_buffer()->RefCountIsOne() && buf_->OwnsMemory(); in RefCountIsOne()
Dtensor.h592 bool RefCountIsOne() const;
Dop_kernel.cc546 if (!input->RefCountIsOne()) { in forward_input()
/external/tensorflow/tensorflow/core/lib/db/
Dsqlite.cc148 DCHECK((*db)->RefCountIsOne()); in Open()
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor_test.cc369 while (!rendez_->RefCountIsOne()) in TEST_F()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc927 if (!op_inputs[i]->RefCountIsOne()) { in EagerKernelExecute()
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc9052 << " is 1?: " << handle->RefCountIsOne(); in getOrCreateSymbolicTensor()
9143 << " is 1?: " << handle->RefCountIsOne(); in TFE_ConsumeInputConcreteTensorFromTraceContext()
Dc_api.cc248 if (buf->RefCountIsOne() && buf->root_buffer()->RefCountIsOne() && in TF_TensorMaybeMove()