| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| H A D | rpc_rendezvous_mgr_test.cc | 151 core::ScopedUnref unref(rendez); in TEST_F()
171 core::ScopedUnref unref(rendez); in TEST_F()
185 core::ScopedUnref unref(rendez); in TEST_F()
205 core::ScopedUnref unref(rendez); in TEST_F()
229 core::ScopedUnref unref(rendez); in TEST_F()
269 core::ScopedUnref unref(rendez); in TEST_F()
302 core::ScopedUnref unref(rendez); in TEST_F()
321 core::ScopedUnref unref(rendez); in TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/platform/ |
| H A D | refcount.h | 85 class ScopedUnref {
87 explicit ScopedUnref(const RefCounted* o) : obj_(o) {} in ScopedUnref() function
88 ~ScopedUnref() { in ~ScopedUnref()
95 ScopedUnref(const ScopedUnref&) = delete;
96 void operator=(const ScopedUnref&) = delete;
|
| H A D | refcount_test.cc | 99 TEST_F(RefTest, ScopedUnref) { in TEST_F() argument
100 { ScopedUnref unref(new MyRef); } in TEST_F()
105 { ScopedUnref unref(nullptr); } in TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | stage_op.cc | 208 core::ScopedUnref scope(buf); in Compute()
230 core::ScopedUnref scope(buf); in Compute()
258 core::ScopedUnref scope(buf); in Compute()
291 core::ScopedUnref scope(buf); in Compute()
315 core::ScopedUnref scope(buf); in Compute()
|
| H A D | map_stage_op.cc | 526 core::ScopedUnref scope(map); in Compute()
582 core::ScopedUnref scope(map); in Compute()
629 core::ScopedUnref scope(map); in Compute()
674 core::ScopedUnref scope(map); in Compute()
724 core::ScopedUnref scope(map); in Compute()
754 core::ScopedUnref scope(map); in Compute()
785 core::ScopedUnref scope(map); in Compute()
|
| H A D | tensor_array_ops.cc | 310 core::ScopedUnref unref(tensor_array); in CreateTensorArray()
433 core::ScopedUnref unref(tensor_array); in Compute()
514 core::ScopedUnref unref(tensor_array); in Compute()
602 core::ScopedUnref unref(tensor_array); in Compute()
826 core::ScopedUnref unref(tensor_array); in Compute()
1030 core::ScopedUnref unref(tensor_array); in Compute()
1225 core::ScopedUnref unref(tensor_array); in Compute()
1391 core::ScopedUnref unref(tensor_array); in Compute()
1436 core::ScopedUnref unref(tensor_array); in Compute()
|
| H A D | stack.cc | 225 core::ScopedUnref unref(stack); in ComputeAsync()
294 core::ScopedUnref unref(stack); in ComputeAsync()
334 core::ScopedUnref unref(stack); in Compute()
|
| H A D | generate_vocab_remapping_op.cc | 65 core::ScopedUnref unref_new(new_vocab_table); in Compute()
97 core::ScopedUnref unref_old(old_vocab_table); in Compute()
|
| H A D | lookup_table_init_op.cc | 55 core::ScopedUnref unref_me(table); in Compute()
126 core::ScopedUnref unref_me(table); in Compute()
|
| H A D | lookup_table_op.cc | 912 core::ScopedUnref unref_me(table); in Compute()
946 core::ScopedUnref unref_me(table); in Compute()
981 core::ScopedUnref unref_me(table); in Compute()
1012 core::ScopedUnref unref_me(table); in Compute()
1033 core::ScopedUnref unref_me(table); in Compute()
1052 core::ScopedUnref unref_me(table); in Compute()
|
| H A D | reader_ops.cc | 85 core::ScopedUnref unref_me(queue); in ComputeWithReader()
119 core::ScopedUnref unref_me(queue); in ComputeWithReader()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| H A D | eager_service_impl.cc | 288 core::ScopedUnref unref_ctx(ctx); in CreateContext()
375 core::ScopedUnref context_unref(server_context); in UpdateContext()
471 core::ScopedUnref context_unref(context); in RunComponentFunction()
600 core::ScopedUnref context_unref(context); in Enqueue()
647 core::ScopedUnref context_unref(context); in WaitQueueDone()
661 core::ScopedUnref context_unref(context); in KeepAlive()
677 core::ScopedUnref context_unref(context); in CloseContext()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| H A D | trt_engine_op_test.cc | 222 core::ScopedUnref sc(cache_resource); in TEST_F()
275 core::ScopedUnref sc(cache_resource); in TEST_F()
304 core::ScopedUnref sc(cache_resource); in TEST_P()
334 core::ScopedUnref sc(cache_resource); in TEST_P()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | iterator_ops.cc | 188 core::ScopedUnref scoped_unref(dataset); in Restore()
606 core::ScopedUnref unref_iterator(iterator_resource); in DoCompute()
806 core::ScopedUnref unref_iterator(*iterator); in TryInit()
919 core::ScopedUnref unref_iterator(iterator); in DoCompute()
959 core::ScopedUnref unref_iterator(iterator); in DoCompute()
1043 core::ScopedUnref unref_iterator(iterator_resource); in Compute()
1082 core::ScopedUnref unref_iterator(iterator_resource); in Compute()
1104 core::ScopedUnref unref_iterator(iterator_resource); in Compute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_compile_op_common.cc | 231 core::ScopedUnref mesh_state_unref(mesh_state); in ComputeInternal()
258 core::ScopedUnref cache_unref(cache); in ComputeInternal()
282 core::ScopedUnref ref_holder_unref(ref_holder); in ComputeInternal()
352 core::ScopedUnref unloader_unref(unloader); in ComputeInternal()
|
| H A D | tpu_configuration_ops.cc | 70 core::ScopedUnref fingerprint_lookup_ref(fingerprint_lookup); in CreateTpuFingerprintLookup()
144 core::ScopedUnref compilation_cache_ref(compilation_cache); in Compute()
215 core::ScopedUnref mesh_state_unref(mesh_state); in Compute()
221 core::ScopedUnref pod_state_unref(pod_state); in Compute()
|
| H A D | tpu_embedding_ops.cc | 64 core::ScopedUnref mesh_state_unref(mesh_state); in Compile()
146 core::ScopedUnref mesh_state_unref(mesh_state); in Compile()
218 core::ScopedUnref mesh_state_unref(mesh_state); in Compile()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| H A D | tpu_compile_ops.cc | 164 core::ScopedUnref mesh_state_unref(mesh_state); in Compute()
191 core::ScopedUnref cache_unref(cache); in Compute()
252 core::ScopedUnref cache_unref(cache); in Compute()
|
| H A D | tpu_execute_op.cc | 69 core::ScopedUnref lookup_unref(proto_lookup); in GetComputationCacheEntry()
325 core::ScopedUnref cache_unref(cache); in DoWork()
428 core::ScopedUnref lookup_unref(proto_lookup); in DoWork()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/summary/ |
| H A D | loader.cc | 80 core::ScopedUnref unref_db(db); in main()
89 core::ScopedUnref unref(db_writer); in main()
|
| H A D | summary_file_writer_test.cc | 55 core::ScopedUnref deleter(writer); in SummaryTestHelper()
270 core::ScopedUnref deleter(writer); in TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | hierarchical_tree_broadcaster_test.cc | 63 core::ScopedUnref unref(cp); \
268 core::ScopedUnref unref(broadcaster); in RunSubdivPermsTest()
351 core::ScopedUnref unref(cp); in TEST_F()
|
| H A D | copy_tensor.cc | 63 core::ScopedUnref status_cb_unref(status_cb); in CopyHostToDevice()
133 core::ScopedUnref status_cb_unref(status_cb); in CopyDeviceToDevice()
347 core::ScopedUnref status_cb_unref(status_cb); in CopyDeviceToHost()
|
| H A D | base_collective_executor.cc | 321 core::ScopedUnref unref(col_impl); in ExecuteAsync()
336 core::ScopedUnref unref(col_impl); in ExecuteAsync()
350 core::ScopedUnref unref(col_impl); in ExecuteAsync()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | lookup_ops.cc | 71 core::ScopedUnref unref(finalized_dataset); in Init()
192 core::ScopedUnref unref_me(table); in ComputeAsync()
|