| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | local_master.cc | 26 Status WaitForNotification(CallOptions* call_options, in WaitForNotification() function
39 n->WaitForNotification(); in WaitForNotification()
43 n->WaitForNotification(); in WaitForNotification()
63 WaitForNotification(call_options, default_timeout_in_ms_, &n)); in CreateSession()
77 WaitForNotification(call_options, default_timeout_in_ms_, &n)); in ExtendSession()
91 WaitForNotification(call_options, default_timeout_in_ms_, &n)); in PartialRunSetup()
106 WaitForNotification(call_options, default_timeout_in_ms_, &n)); in RunStep()
128 WaitForNotification(call_options, default_timeout_in_ms_, &n)); in CloseSession()
142 WaitForNotification(call_options, default_timeout_in_ms_, &n)); in ListDevices()
156 WaitForNotification(call_options, default_timeout_in_ms_, &n)); in Reset()
[all …]
|
| D | collective_rma_distributed_test.cc | 287 consumer_note.WaitForNotification(); in TEST_F()
289 producer_note.WaitForNotification(); in TEST_F()
322 consumer_note.WaitForNotification(); in TEST_F()
324 producer_note.WaitForNotification(); in TEST_F()
348 consumer_note.WaitForNotification(); in TEST_F()
|
| D | rpc_collective_executor_mgr_test.cc | 116 note.WaitForNotification(); in TEST_F()
142 note.WaitForNotification(); in TEST_F()
162 note.WaitForNotification(); in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | shared_batch_scheduler_test.cc | 60 start->WaitForNotification(); in CreateFakeClockAdvancerThread()
233 first_batch_processed.WaitForNotification(); in TEST()
242 second_batch_processed.WaitForNotification(); in TEST()
250 third_batch_processed.WaitForNotification(); in TEST()
285 first_batch_processed.WaitForNotification(); in TEST()
289 second_batch_processed.WaitForNotification(); in TEST()
329 first_batch_processed.WaitForNotification(); in TEST()
331 second_batch_processed.WaitForNotification(); in TEST()
354 queue_0_first_batch_proceed.WaitForNotification(); in TEST()
365 queue_1_first_batch_proceed.WaitForNotification(); in TEST()
[all …]
|
| D | serial_device_batch_scheduler_test.cc | 59 start->WaitForNotification(); in CreateFakeClockAdvancerThread()
116 finish_processing.WaitForNotification(); in TEST()
161 start_processing.WaitForNotification(); in TEST()
272 finish_processing.WaitForNotification(); in TEST()
323 start_processing.WaitForNotification(); in TEST()
344 finish_processing.WaitForNotification(); in TEST()
364 finish_processing.WaitForNotification(); in TEST()
|
| D | adaptive_shared_batch_scheduler_test.cc | 59 start->WaitForNotification(); in CreateFakeClockAdvancerThread()
119 finish_processing.WaitForNotification(); in TEST()
212 finish_processing.WaitForNotification(); in TEST()
277 finish_processing.WaitForNotification(); in TEST()
317 finish_processing.WaitForNotification(); in TEST()
357 finish_processing.WaitForNotification(); in TEST()
|
| D | batch_scheduler_test.cc | 105 do_delete.WaitForNotification(); in TEST()
113 deleted.WaitForNotification(); in TEST()
|
| D | periodic_function_test.cc | 128 listener->WaitForNotification(); in TEST()
137 listener->WaitForNotification(); in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | buf_rendezvous_test.cc | 72 note.WaitForNotification(); in TEST_F()
103 note.WaitForNotification(); in TEST_F()
122 note.WaitForNotification(); in TEST_F()
160 prod_note.WaitForNotification(); in TEST_F()
161 cons_note.WaitForNotification(); in TEST_F()
188 prod_note.WaitForNotification(); in TEST_F()
189 cons_note.WaitForNotification(); in TEST_F()
|
| D | collective_rma_local_test.cc | 92 recv_note.WaitForNotification(); in TEST_F()
93 send_note.WaitForNotification(); in TEST_F()
137 recv_note.WaitForNotification(); in TEST_F()
138 send_note.WaitForNotification(); in TEST_F()
|
| D | ring_reducer.cc | 44 RingReducer::~RingReducer() { group_size_tensor_ready_.WaitForNotification(); } in ~RingReducer()
100 note.WaitForNotification(); in Run()
182 note.WaitForNotification(); in RunAsyncParts()
248 group_size_tensor_ready_.WaitForNotification(); in RunAsyncParts()
|
| D | device_resolver_local_test.cc | 62 note.WaitForNotification(); in TEST_F()
80 note.WaitForNotification(); in TEST_F()
|
| D | rendezvous_util_test.cc | 64 n.WaitForNotification(); in TEST_F()
83 n.WaitForNotification(); in TEST_F()
|
| D | collective_executor_mgr_test.cc | 81 ss_note.WaitForNotification(); in TEST_F()
94 gs_note.WaitForNotification(); in TEST_F()
|
| D | collective_param_resolver_local_test.cc | 185 note[i].WaitForNotification(); in TEST_F()
238 note[i].WaitForNotification(); in TEST_F()
276 note[i].WaitForNotification(); in TEST_F()
|
| D | ring_gatherer.cc | 111 note.WaitForNotification(); in Run()
146 note.WaitForNotification(); in RunAsyncParts()
|
| /external/tensorflow/tensorflow/core/lib/core/ |
| D | notification_test.cc | 35 proceed.WaitForNotification(); in TEST()
40 start.WaitForNotification(); in TEST()
65 n.WaitForNotification(); in TEST()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | cancellation_test.cc | 113 done[i].WaitForNotification(); in TEST()
151 finish_callback.WaitForNotification(); in TEST()
160 cancel_started.WaitForNotification(); in TEST()
166 cancel_complete.WaitForNotification(); in TEST()
|
| /external/tensorflow/tensorflow/cc/training/ |
| D | coordinator_test.cc | 47 about_to_wait.WaitForNotification(); in TEST()
52 done.WaitForNotification(); in TEST()
100 if (start != nullptr) start->WaitForNotification(); in CountThread()
109 start->WaitForNotification(); in SetStatusThread()
|
| /external/google-breakpad/src/testing/gtest/test/ |
| D | gtest_output_test_.cc | 275 check_points->n2.WaitForNotification(); in ThreadWithScopedTrace()
292 check_points.n1.WaitForNotification(); in TEST()
299 check_points.n3.WaitForNotification(); in TEST()
476 notifications->spawn_thread_ok_to_terminate.WaitForNotification(); in ThreadRoutine()
488 notifications_.spawn_thread_started.WaitForNotification(); in SetUp()
|
| /external/googletest/googletest/test/ |
| D | googletest-output-test_.cc | 295 check_points->n2.WaitForNotification(); in ThreadWithScopedTrace()
311 check_points.n1.WaitForNotification(); in TEST()
318 check_points.n3.WaitForNotification(); in TEST()
494 notifications->spawn_thread_ok_to_terminate.WaitForNotification(); in ThreadRoutine()
506 notifications_.spawn_thread_started.WaitForNotification(); in SetUp()
|
| /external/tensorflow/tensorflow/core/util/rpc/ |
| D | call_container.h | 120 callback_destroyed_->WaitForNotification(); in CallContainer()
136 calls_started->WaitForNotification(); in CallContainer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_transfer_manager.cc | 68 StatusOr<Shape> WaitForNotification() { in WaitForNotification() function in xla::__anon79616a0d0111::CpuOutfeedBuffer
69 done_.WaitForNotification(); in WaitForNotification()
276 TF_ASSIGN_OR_RETURN(Shape outfed_shape, buffer->WaitForNotification()); in TransferBuffersFromOutfeedInternal()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer.cc | 190 group_size_ready.WaitForNotification(); in Run()
191 nccl_done.WaitForNotification(); in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | transfer_manager.cc | 67 n.WaitForNotification(); in TransferLiteralFromDevice()
91 n.WaitForNotification(); in TransferLiteralFromDevice()
132 n.WaitForNotification(); in TransferArrayFromDevice()
|