| /external/tensorflow/tensorflow/core/platform/ |
| D | refcount.h | 49 bool RefCountIsOne() const;
101 if (RefCountIsOne() || ref_.fetch_sub(1) == 1) { in Unref()
111 inline bool RefCounted::RefCountIsOne() const { in RefCountIsOne() function
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_resource_ops_test.cc | 164 EXPECT_FALSE(resource->RefCountIsOne()); in TEST_F()
180 EXPECT_TRUE(resource->RefCountIsOne()); in TEST_F()
228 EXPECT_FALSE(resource->RefCountIsOne()); in TEST_F()
244 EXPECT_TRUE(resource->RefCountIsOne()); in TEST_F()
|
| /external/tensorflow/tensorflow/core/lib/core/ |
| D | refcount_test.cc | 64 ASSERT_TRUE(ref->RefCountIsOne()); in TEST_F()
71 ASSERT_FALSE(ref->RefCountIsOne()); in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tensor_list.h | 142 bool RefCountIsOne() const { return tensors_->RefCountIsOne(); } in RefCountIsOne() function
|
| D | tensor_map.h | 163 bool RefCountIsOne() const { return tensors_->RefCountIsOne(); } in RefCountIsOne() function
|
| D | training_op_helpers.h | 40 if (var->tensor()->RefCountIsOne()) { in EnsureSparseVariableAccess()
200 if (copy_on_read_mode || !tensor->RefCountIsOne()) { in PrepareToUpdateVariable()
|
| D | list_kernels.cc | 109 if (tmp_out->RefCountIsOne()) { in ForwardInputOrCreateNewList()
350 if (out->RefCountIsOne()) { in Compute()
479 if (aliased == nullptr || !aliased->RefCountIsOne()) { in Compute()
|
| D | map_kernels.h | 59 if (tmp_out->RefCountIsOne()) { in ForwardInputOrCreateNewMap()
|
| D | resource_variable_ops.cc | 471 if (!variable->tensor()->RefCountIsOne() || in Compute()
|
| D | list_kernels.h | 922 if (tl_i == nullptr || !tl_i->RefCountIsOne()) { in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | batch_util.cc | 55 if (element.RefCountIsOne()) { in HandleElementToSlice()
68 if (element.RefCountIsOne()) { in HandleElementToSlice()
135 if (parent->RefCountIsOne()) { in HandleSliceToElement()
147 if (parent->RefCountIsOne()) { in HandleSliceToElement()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute_node.h | 115 if (h->RefCountIsOne()) { in Run()
193 if (h->RefCountIsOne()) { in Run()
|
| D | context.cc | 747 is_first_ref = registered_function->RefCountIsOne(); in AddFunctionDef()
776 is_last_ref = registered_function->RefCountIsOne(); in RemoveFunction()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_compilation_cache.cc | 72 while (!entry.second->RefCountIsOne()) { in ~XRTCompilationCache()
111 if (entry->RefCountIsOne()) { in DiscardEntryRefLocked()
|
| D | xrt_state.cc | 562 !index_buffer.second->RefCountIsOne()) { in IsExclusiveOwner()
|
| /external/tensorflow/tensorflow/c/ |
| D | tf_tensor.cc | 192 if (buf->RefCountIsOne() && buf->root_buffer()->RefCountIsOne() && in CanMove()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/eager/ |
| D | grpc_eager_service_impl.h | 124 if (call->RefCountIsOne()) { in StreamingEnqueueHandler()
|
| D | grpc_eager_client.cc | 86 if (RefCountIsOne()) { in GrpcEagerClientThread()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compilation_cache_interface.cc | 221 CHECK(entry->RefCountIsOne()); in UnloadAndDestroy()
246 if (entry->RefCountIsOne()) { in DiscardEntryRef()
|
| D | tpu_execute_op.cc | 329 bool can_reuse_buffers = tensor.RefCountIsOne() && may_reuse; in BuildComputationInputs()
|
| /external/tensorflow/tensorflow/core/lib/db/ |
| D | sqlite.cc | 148 DCHECK((*db)->RefCountIsOne()); in Open()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.cc | 111 if (RefCountIsOne()) { in FillAllocationDescription()
712 bool Tensor::RefCountIsOne() const { in RefCountIsOne() function in tensorflow::Tensor
713 return buf_ != nullptr && buf_->RefCountIsOne() && in RefCountIsOne()
714 buf_->root_buffer()->RefCountIsOne() && buf_->OwnsMemory(); in RefCountIsOne()
|
| D | tensor.h | 655 bool RefCountIsOne() const;
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 252 t->RefCountIsOne() && is_updated_resource_variable && in PopulateInputs()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | executor_test.cc | 381 while (!rendez_->RefCountIsOne()) { in TEST_F()
|