| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| local_master.cc | 26 Status WaitForNotification(CallOptions* call_options,
39 n->WaitForNotification();
43 n->WaitForNotification();
63 WaitForNotification(call_options, default_timeout_in_ms_, &n));
77 WaitForNotification(call_options, default_timeout_in_ms_, &n));
91 WaitForNotification(call_options, default_timeout_in_ms_, &n));
106 WaitForNotification(call_options, default_timeout_in_ms_, &n));
128 WaitForNotification(call_options, default_timeout_in_ms_, &n));
142 WaitForNotification(call_options, default_timeout_in_ms_, &n));
156 WaitForNotification(call_options, default_timeout_in_ms_, &n))
[all...] |
| collective_rma_distributed_test.cc | 287 consumer_note.WaitForNotification();
289 producer_note.WaitForNotification();
322 consumer_note.WaitForNotification();
324 producer_note.WaitForNotification();
348 consumer_note.WaitForNotification();
|
| remote_device_test.cc | 71 n.WaitForNotification();
|
| rpc_collective_executor_mgr_test.cc | 116 note.WaitForNotification();
142 note.WaitForNotification();
162 note.WaitForNotification();
|
| /external/tensorflow/tensorflow/core/lib/core/ |
| notification_test.cc | 35 proceed.WaitForNotification();
40 start.WaitForNotification();
65 n.WaitForNotification();
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| outfeed_manager.h | 39 void WaitUntilAvailable() { done_.WaitForNotification(); }
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| shared_batch_scheduler_test.cc | 60 start->WaitForNotification();
233 first_batch_processed.WaitForNotification();
242 second_batch_processed.WaitForNotification();
250 third_batch_processed.WaitForNotification();
285 first_batch_processed.WaitForNotification();
289 second_batch_processed.WaitForNotification();
329 first_batch_processed.WaitForNotification();
331 second_batch_processed.WaitForNotification();
354 queue_0_first_batch_proceed.WaitForNotification();
365 queue_1_first_batch_proceed.WaitForNotification();
[all...] |
| fake_clock_env.cc | 85 wake_notification.WaitForNotification();
|
| serial_device_batch_scheduler_test.cc | 59 start->WaitForNotification();
116 finish_processing.WaitForNotification();
161 start_processing.WaitForNotification();
272 finish_processing.WaitForNotification();
323 start_processing.WaitForNotification();
344 finish_processing.WaitForNotification();
364 finish_processing.WaitForNotification();
|
| adaptive_shared_batch_scheduler_test.cc | 59 start->WaitForNotification();
119 finish_processing.WaitForNotification();
212 finish_processing.WaitForNotification();
277 finish_processing.WaitForNotification();
317 finish_processing.WaitForNotification();
357 finish_processing.WaitForNotification();
|
| batch_scheduler_test.cc | 105 do_delete.WaitForNotification();
113 deleted.WaitForNotification();
|
| periodic_function_test.cc | 128 listener->WaitForNotification();
137 listener->WaitForNotification();
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| device_resolver_local_test.cc | 62 note.WaitForNotification();
80 note.WaitForNotification();
|
| rendezvous_util_test.cc | 64 n.WaitForNotification();
83 n.WaitForNotification();
|
| buf_rendezvous_test.cc | 72 note.WaitForNotification();
103 note.WaitForNotification();
122 note.WaitForNotification();
160 prod_note.WaitForNotification();
161 cons_note.WaitForNotification();
188 prod_note.WaitForNotification();
189 cons_note.WaitForNotification();
|
| collective_rma_local_test.cc | 92 recv_note.WaitForNotification();
93 send_note.WaitForNotification();
137 recv_note.WaitForNotification();
138 send_note.WaitForNotification();
|
| collective_executor_mgr_test.cc | 81 ss_note.WaitForNotification();
94 gs_note.WaitForNotification();
|
| ring_reducer.cc | 44 RingReducer::~RingReducer() { group_size_tensor_ready_.WaitForNotification(); }
100 note.WaitForNotification();
182 note.WaitForNotification();
248 group_size_tensor_ready_.WaitForNotification();
|
| /external/tensorflow/tensorflow/core/framework/ |
| cancellation_test.cc | 113 done[i].WaitForNotification();
151 finish_callback.WaitForNotification();
160 cancel_started.WaitForNotification();
166 cancel_complete.WaitForNotification();
|
| cancellation.cc | 89 cancelled_notification_.WaitForNotification();
|
| /external/tensorflow/tensorflow/core/util/rpc/ |
| call_container.h | 120 callback_destroyed_->WaitForNotification();
136 calls_started->WaitForNotification();
|
| /external/tensorflow/tensorflow/cc/training/ |
| coordinator_test.cc | 47 about_to_wait.WaitForNotification();
52 done.WaitForNotification();
100 if (start != nullptr) start->WaitForNotification();
109 start->WaitForNotification();
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| remote_execute_node.h | 69 n.WaitForNotification();
|
| /external/tensorflow/tensorflow/core/platform/default/ |
| notification.h | 50 void WaitForNotification() {
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| unbounded_thread_pool.cc | 57 join_notification_->WaitForNotification();
|
