Lines Matching refs:os_nb
556 for (int os_nb = 0; os_nb < jcp.os_nb_block; ++os_nb) { in execute_backward_data_ncsp() local
557 auto out_off = os_nb * m + od * jcp.os; in execute_backward_data_ncsp()
561 = nstl::min((dim_t)jcp.os_block, jcp.os - os_nb * m); in execute_backward_data_ncsp()
576 os_nb * jcp.os_block, os_block); in execute_backward_data_ncsp()
579 _diff_src, od, os_nb * jcp.os_block, os_block); in execute_backward_data_ncsp()
824 for (int os_nb = 0; os_nb < jcp.os_nb_block; ++os_nb) { in execute_backward_weights_ncsp() local
825 auto out_off = os_nb * k + od * jcp.os; in execute_backward_weights_ncsp()
827 (dim_t)jcp.os_block, jcp.os - os_nb * k); in execute_backward_weights_ncsp()
834 _src, _col, os_nb * jcp.os_block, in execute_backward_weights_ncsp()
839 os_nb * jcp.os_block, os_block); in execute_backward_weights_ncsp()
847 mb == mb_start && os_nb == 0 && od == 0 ? &zero in execute_backward_weights_ncsp()
856 os_nb = jcp.os_nb_block; in execute_backward_weights_ncsp()