Searched refs:event_after_add (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr_test.cc | 373 void DoAddChain(int adds_per_copy, int rounds, bool event_after_add, in DoAddChain() argument
406 if (event_after_add) { in DoAddChain()
480 bool event_after_add, int pending_cap) { in BM_chain_ops() argument
483 bool event_after_add, int pending_cap, int threads) { in BM_chain_ops()
518 int expected = 1 + (event_after_add ? adds_per_round : 0); in BM_chain_ops()
519 bm_helper->DoAddChain(adds_per_round, 1, event_after_add, callback, nullptr); in BM_chain_ops()
526 expected = iters * (1 + (event_after_add ? adds_per_round : 0)); in BM_chain_ops()
527 bm_helper->DoAddChain(adds_per_round, iters, event_after_add, callback, in BM_chain_ops()
530 expected = threads * iters * (1 + (event_after_add ? adds_per_round : 0)); in BM_chain_ops()
533 [callback, iters, adds_per_round, event_after_add, time_ptr]() { in BM_chain_ops()
[all …]
|