Searched refs:adds_per_round (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr_test.cc | 479 static void BM_chain_ops(int iters, int tensor_size, int adds_per_round, in BM_chain_ops() argument
482 static void BM_chain_ops(int iters, int tensor_size, int adds_per_round, in BM_chain_ops()
503 if (bm_helper->num_ops() != adds_per_round || in BM_chain_ops()
505 bm_helper->ReInit(adds_per_round, tensor_size); 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 …]
|