Searched defs:is_os_tail (Results 1 – 6 of 6) sorted by relevance
263 const bool is_os_tail = jcp.os - os < jcp.os_block; in maybe_rtus() local339 const bool is_os_tail = jcp.is_os_blocking ? (jcp.os - os < jcp.os_block) in exec_ker() local
113 bool is_os_tail = (jbgp.mb < jbgp.os_block); in execute_forward() local162 bool is_os_tail = (jbgp.mb - n < jbgp.os_block); in execute_forward() local394 const bool is_os_tail = cur_os_block < jbgp.os_block; in execute_forward() local515 bool is_os_tail = (jbgp.mb < jbgp.os_block); in execute_backward_data() local622 const bool is_os_tail = (jbgp.mb - n < jbgp.os_block); in execute_backward_data() local1186 bool is_os_tail = jbgp.mb - n < jbgp.os_block * jbgp.nb_os_blocking; in compute_diff_weights_and_bias() local
251 auto compute_M = [=](bool is_os_tail) { in generate()284 auto compute_batch = [=](bool is_os_tail) { in generate()594 bool is_os_tail) { in generate()
111 bool is_os_tail = (jbgp.mb < jbgp.os_block); in execute_forward() local160 bool is_os_tail = (jbgp.mb - n < jbgp.os_block); in execute_forward() local391 const bool is_os_tail = cur_os_block < jbgp.os_block; in execute_forward() local510 bool is_os_tail = (jbgp.mb < jbgp.os_block); in execute_backward_data() local617 const bool is_os_tail = (jbgp.mb - n < jbgp.os_block); in execute_backward_data() local1151 bool is_os_tail = jbgp.mb - n < jbgp.os_block * jbgp.nb_os_blocking; in compute_diff_weights_and_bias() local
231 const bool is_os_tail = jcp.is_os_blocking ? (jcp.os - os < jcp.os_block) in exec_ker() local
249 auto compute_M = [=](bool is_os_tail) { in generate()282 auto compute_batch = [=](bool is_os_tail) { in generate()592 bool is_os_tail) { in generate()