Lines Matching refs:jpp

171             const jit_pool_conf_t &jpp, std::size_t offset_multiplier = 1u)  in transpose_ncsp_to_block_fmt_t()
174 , c_without_padding_(jpp.c_without_padding) in transpose_ncsp_to_block_fmt_t()
175 , c_block_(jpp.c_block) in transpose_ncsp_to_block_fmt_t()
213 const jit_pool_conf_t &jpp, std::size_t offset_multiplier = 1u) in transpose_block_fmt_to_ncsp_t()
216 , c_without_padding_(jpp.c_without_padding) in transpose_block_fmt_to_ncsp_t()
217 , c_block_(jpp.c_block) in transpose_block_fmt_to_ncsp_t()
251 transpose_facade_base_t(const jit_pool_conf_t &jpp, in transpose_facade_base_t() argument
255 : src_sp_(static_cast<dim_t>(jpp.id) * jpp.ih * jpp.iw) in transpose_facade_base_t()
256 , dst_sp_(static_cast<dim_t>(jpp.od) * jpp.oh * jpp.ow) in transpose_facade_base_t()
257 , src_slice_(src_sp_ * jpp.c_block) in transpose_facade_base_t()
258 , dst_slice_(dst_sp_ * jpp.c_block) in transpose_facade_base_t()
259 , transpose_src_(jpp.tag_kind == jit_memory_tag_kind_t::ncsp) in transpose_facade_base_t()
260 , transpose_dst_(jpp.tag_kind == jit_memory_tag_kind_t::ncsp) in transpose_facade_base_t()
290 std::size_t ithr, int ih, const jit_pool_conf_t &jpp) const { in get_src_addr()
292 return static_cast<const void *>(&wsp[ih * jpp.iw * jpp.c_block]); in get_src_addr()
296 std::size_t ithr, int oh, const jit_pool_conf_t &jpp) const { in get_dst_addr()
298 return static_cast<const void *>(&wsp[oh * jpp.ow * jpp.c_block]); in get_dst_addr()
302 std::size_t ithr, int oh, const jit_pool_conf_t &jpp) const { in get_indices_addr()
306 &wsp[oh * jpp.ow * jpp.c_block * ind_dt_size_]); in get_indices_addr()
310 const jit_pool_conf_t &jpp) const { in get_src_addr_3d()
312 return static_cast<const void *>(&wsp[ih * jpp.iw * jpp.c_block in get_src_addr_3d()
313 + id * jpp.ih * jpp.iw * jpp.c_block]); in get_src_addr_3d()
317 const jit_pool_conf_t &jpp) const { in get_dst_addr_3d()
319 return static_cast<const void *>(&wsp[oh * jpp.ow * jpp.c_block in get_dst_addr_3d()
320 + od * jpp.oh * jpp.ow * jpp.c_block]); in get_dst_addr_3d()
324 const jit_pool_conf_t &jpp) const { in get_indices_addr_3d()
328 &wsp[oh * jpp.ow * jpp.c_block * ind_dt_size_ in get_indices_addr_3d()
329 + od * jpp.oh * jpp.ow * jpp.c_block * ind_dt_size_]); in get_indices_addr_3d()
366 fwd_pooling_transpose_facade_t(const jit_pool_conf_t &jpp, in fwd_pooling_transpose_facade_t() argument
373 jpp, src_d, dst_d, indices_d, indices, wsp_dt, ctx) { in fwd_pooling_transpose_facade_t()
380 this->cvt_slice_src_wsp_, this->src_slice_, jpp); in fwd_pooling_transpose_facade_t()
398 this->dst_d_, jpp, 1u), in fwd_pooling_transpose_facade_t()
403 this->indices_d_, jpp, this->ind_dt_size_), in fwd_pooling_transpose_facade_t()
413 bwd_pooling_transpose_facade_t(const jit_pool_conf_t &jpp, in bwd_pooling_transpose_facade_t() argument
420 jpp, src_d, dst_d, indices_d, indices, wsp_dt, ctx) in bwd_pooling_transpose_facade_t()
421 , c_tail_(jpp.c_without_padding % jpp.c_block) { in bwd_pooling_transpose_facade_t()
429 this->src_d_, jpp, 1u); in bwd_pooling_transpose_facade_t()
446 this->cvt_slice_dst_wsp_, this->dst_slice_, jpp), in bwd_pooling_transpose_facade_t()
451 this->dst_slice_, jpp, this->ind_dt_size_), in bwd_pooling_transpose_facade_t()
461 std::size_t ithr, const jit_pool_conf_t &jpp) const { in fill_input_c_tail_with_zeros()
466 for (dim_t c = c_tail_; c < jpp.c_block; c++) in fill_input_c_tail_with_zeros()
467 wsp_ptr[s * jpp.c_block + c] = 0.f; in fill_input_c_tail_with_zeros()
472 for_(dim_t c = c_tail_; c < jpp.c_block; c++) in fill_input_c_tail_with_zeros()
474 ind_ptr[(s * jpp.c_block + c) * this->ind_dt_size_ + i] = 0; in fill_input_c_tail_with_zeros()
504 const auto &jpp = pd()->jpp_; in init_ncsp_trans_ctx() local
506 const dim_t src_sp = static_cast<dim_t>(jpp.id) * jpp.ih * jpp.iw; in init_ncsp_trans_ctx()
507 const dim_t dst_sp = static_cast<dim_t>(jpp.od) * jpp.oh * jpp.ow; in init_ncsp_trans_ctx()
508 const auto res = std::div(jpp.c_without_padding, jpp.c_block); in init_ncsp_trans_ctx()
517 d_type, src_sp, wsp_dt, jpp.c_block, jpp.c_block, src_sp); in init_ncsp_trans_ctx()
519 wsp_dt, jpp.c_block, d_type, dst_sp, dst_sp, jpp.c_block); in init_ncsp_trans_ctx()
522 indices_d.data_type(), jpp.c_block, indices_d.data_type(), in init_ncsp_trans_ctx()
523 dst_sp, dst_sp, jpp.c_block); in init_ncsp_trans_ctx()
528 d_type, src_sp, wsp_dt, jpp.c_block, c_tail, src_sp); in init_ncsp_trans_ctx()
530 wsp_dt, jpp.c_block, d_type, dst_sp, dst_sp, c_tail); in init_ncsp_trans_ctx()
533 indices_d.data_type(), jpp.c_block, indices_d.data_type(), in init_ncsp_trans_ctx()
552 const auto &jpp = pd()->jpp_; in execute_forward() local
558 = fwd_pooling_transpose_facade_t<data_t, wsp_data_t, d_type>(jpp, in execute_forward()
566 assert(ur_bc == jpp.ur_bc || ur_bc == jpp.ur_bc_tail); in execute_forward()
569 const int ij = oh * jpp.stride_h; in execute_forward()
570 const int i_t_overflow = nstl::max(0, jpp.t_pad - ij); in execute_forward()
572 = nstl::max(jpp.ih, ij + jpp.kh - jpp.t_pad) - jpp.ih; in execute_forward()
573 const int ih = nstl::max(ij - jpp.t_pad, 0); in execute_forward()
576 = ((jpp.tag_kind == jit_memory_tag_kind_t::nspc) ? jpp.c_block in execute_forward()
579 const int c_elem_off = jpp.c_block * b_c; in execute_forward()
582 arg.src = transpose_facade.get_src_addr(ithr, ih, jpp); in execute_forward()
588 arg.dst = transpose_facade.get_dst_addr(ithr, oh, jpp); in execute_forward()
595 arg.indices = transpose_facade.get_indices_addr(ithr, oh, jpp); in execute_forward()
602 arg.kh_padding = jpp.kh - i_t_overflow - i_b_overflow; in execute_forward()
603 arg.kh_padding_shift = i_t_overflow * jpp.kw; in execute_forward()
604 arg.ker_area_h = static_cast<float>(jpp.kh in execute_forward()
605 - nstl::max(0, oh * jpp.stride_h - jpp.t_pad + jpp.kh - jpp.ih) in execute_forward()
606 - nstl::max(0, jpp.t_pad - oh * jpp.stride_h)); in execute_forward()
613 if (jpp.tag_kind == jit_memory_tag_kind_t::nspc) { in execute_forward()
614 const auto nb2_c = utils::div_up(jpp.nb_c, jpp.ur_bc); in execute_forward()
615 parallel_nd(jpp.mb, jpp.oh, nb2_c, [&](int n, int oh, int b2_c) { in execute_forward()
616 const auto b_c = b2_c * jpp.ur_bc; in execute_forward()
617 const auto ur_bc = nstl::min(jpp.ur_bc, jpp.nb_c - b_c); in execute_forward()
623 parallel_nd_ext(0, jpp.mb, jpp.nb_c, in execute_forward()
628 for (int oh = 0; oh < jpp.oh; ++oh) in execute_forward()
638 = static_cast<std::size_t>(jpp.mb) * jpp.nb_c * jpp.oh; in execute_forward()
646 start, n, jpp.mb, b_c, jpp.nb_c, oh, jpp.oh); in execute_forward()
651 n, jpp.mb, b_c, jpp.nb_c, oh, jpp.oh); in execute_forward()
662 const auto &jpp = pd()->jpp_; in execute_forward_3d() local
674 = fwd_pooling_transpose_facade_t<data_t, wsp_data_t, d_type>(jpp, in execute_forward_3d()
683 assert(ur_bc == jpp.ur_bc || ur_bc == jpp.ur_bc_tail); in execute_forward_3d()
686 const int ij = oh * jpp.stride_h; in execute_forward_3d()
687 const int i_t_overflow = nstl::max(0, jpp.t_pad - ij); in execute_forward_3d()
689 = nstl::max(jpp.ih, ij + jpp.kh - jpp.t_pad) - jpp.ih; in execute_forward_3d()
690 const int ih = nstl::max(ij - jpp.t_pad, 0); in execute_forward_3d()
692 = ((jpp.tag_kind == jit_memory_tag_kind_t::nspc) ? jpp.c_block in execute_forward_3d()
697 arg.src = transpose_facade.get_src_addr_3d(ithr, id, ih, jpp); in execute_forward_3d()
702 arg.dst = transpose_facade.get_dst_addr_3d(ithr, od, oh, jpp); in execute_forward_3d()
709 ithr, od, oh, jpp); in execute_forward_3d()
716 arg.kd_padding = jpp.kd - d_t_overflow - d_b_overflow; in execute_forward_3d()
717 arg.kh_padding = jpp.kh - i_t_overflow - i_b_overflow; in execute_forward_3d()
719 = i_t_overflow * jpp.kw + d_t_overflow * jpp.kw * jpp.kh; in execute_forward_3d()
720 arg.kd_padding_shift = (i_t_overflow + i_b_overflow) * jpp.kw; in execute_forward_3d()
721 arg.ker_area_h = (float)(jpp.kh in execute_forward_3d()
723 oh * jpp.stride_h - jpp.t_pad + jpp.kh in execute_forward_3d()
724 - jpp.ih) in execute_forward_3d()
725 - nstl::max(0, jpp.t_pad - oh * jpp.stride_h)) in execute_forward_3d()
726 * (jpp.kd in execute_forward_3d()
728 od * jpp.stride_d - jpp.f_pad + jpp.kd - jpp.id) in execute_forward_3d()
729 - nstl::max(0, jpp.f_pad - od * jpp.stride_d)); in execute_forward_3d()
733 arg.c_elem_off = jpp.c_block * b_c; in execute_forward_3d()
737 if (jpp.tag_kind == jit_memory_tag_kind_t::nspc) { in execute_forward_3d()
738 const auto nb2_c = utils::div_up(jpp.nb_c, jpp.ur_bc); in execute_forward_3d()
739 parallel_nd(jpp.mb, jpp.od, nb2_c, [&](int n, int od, int b2_c) { in execute_forward_3d()
740 const auto b_c = b2_c * jpp.ur_bc; in execute_forward_3d()
741 const auto ur_bc = nstl::min(jpp.ur_bc, jpp.nb_c - b_c); in execute_forward_3d()
743 const int ik = od * jpp.stride_d; in execute_forward_3d()
744 const int d_t_overflow = nstl::max(0, jpp.f_pad - ik); in execute_forward_3d()
746 = nstl::max(jpp.id, ik + jpp.kd - jpp.f_pad) - jpp.id; in execute_forward_3d()
747 const int id = nstl::max(ik - jpp.f_pad, 0); in execute_forward_3d()
748 for (int oh = 0; oh < jpp.oh; ++oh) { in execute_forward_3d()
755 parallel_nd_ext(0, jpp.mb, jpp.nb_c, in execute_forward_3d()
761 for (int od = 0; od < jpp.od; ++od) { in execute_forward_3d()
762 const int ik = od * jpp.stride_d; in execute_forward_3d()
764 = nstl::max(0, jpp.f_pad - ik); in execute_forward_3d()
766 = nstl::max(jpp.id, ik + jpp.kd - jpp.f_pad) in execute_forward_3d()
767 - jpp.id; in execute_forward_3d()
768 const int id = nstl::max(ik - jpp.f_pad, 0); in execute_forward_3d()
769 for (int oh = 0; oh < jpp.oh; ++oh) { in execute_forward_3d()
780 parallel_nd(jpp.mb, jpp.nb_c, jpp.od, [&](int n, int b_c, int od) { in execute_forward_3d()
781 const int ik = od * jpp.stride_d; in execute_forward_3d()
782 const int d_t_overflow = nstl::max(0, jpp.f_pad - ik); in execute_forward_3d()
784 = nstl::max(jpp.id, ik + jpp.kd - jpp.f_pad) - jpp.id; in execute_forward_3d()
785 const int id = nstl::max(ik - jpp.f_pad, 0); in execute_forward_3d()
786 for (int oh = 0; oh < jpp.oh; ++oh) { in execute_forward_3d()
810 const auto &jpp = pd()->jpp_; in init_ncsp_trans_ctx() local
812 const dim_t diff_src_sp = static_cast<dim_t>(jpp.id) * jpp.ih * jpp.iw; in init_ncsp_trans_ctx()
813 const dim_t diff_dst_sp = static_cast<dim_t>(jpp.od) * jpp.oh * jpp.ow; in init_ncsp_trans_ctx()
814 const auto res = std::div(jpp.c_without_padding, jpp.c_block); in init_ncsp_trans_ctx()
823 diff_dst_sp, wsp_dt, jpp.c_block, jpp.c_block, diff_dst_sp); in init_ncsp_trans_ctx()
825 jpp.c_block, d_type, diff_src_sp, diff_src_sp, jpp.c_block); in init_ncsp_trans_ctx()
829 jpp.c_block, jpp.c_block, diff_dst_sp); in init_ncsp_trans_ctx()
833 d_type, diff_dst_sp, wsp_dt, jpp.c_block, c_tail, diff_dst_sp); in init_ncsp_trans_ctx()
835 wsp_dt, jpp.c_block, d_type, diff_src_sp, diff_src_sp, c_tail); in init_ncsp_trans_ctx()
839 jpp.c_block, c_tail, diff_dst_sp); in init_ncsp_trans_ctx()
865 const auto &jpp = pd()->jpp_; in execute_backward() local
868 wsp_data_t, d_type>(jpp, trans_ctx_.get(), diff_src_d, in execute_backward()
873 return nstl::min(nstl::max(oh * jpp.stride_h - jpp.t_pad, 0), jpp.ih); in execute_backward()
878 nstl::max(oh * jpp.stride_h - jpp.t_pad + jpp.kh, 0), jpp.ih); in execute_backward()
887 const auto c_off = jpp.is_plain() ? b_c * jpp.c_block : b_c; in execute_backward()
889 arg.src = transpose_facade.get_src_addr(ithr, ih, jpp); in execute_backward()
894 arg.dst = transpose_facade.get_dst_addr(ithr, oh, jpp); in execute_backward()
900 arg.indices = transpose_facade.get_indices_addr(ithr, oh, jpp); in execute_backward()
909 const int zero_ih_end = (oh == jpp.oh - 1) ? jpp.ih : get_last_ih(oh); in execute_backward()
915 = transpose_facade.get_src_addr(ithr, zero_ih_start, jpp); in execute_backward()
920 const int i_t_overflow = nstl::max(0, jpp.t_pad - oh * jpp.stride_h); in execute_backward()
922 = nstl::max(jpp.ih, oh * jpp.stride_h + jpp.kh - jpp.t_pad) in execute_backward()
923 - jpp.ih; in execute_backward()
924 arg.kh_padding = jpp.kh - i_t_overflow - i_b_overflow; in execute_backward()
925 arg.kh_padding_shift = i_t_overflow * jpp.kw; in execute_backward()
926 arg.ker_area_h = static_cast<float>(jpp.kh in execute_backward()
927 - nstl::max(0, oh * jpp.stride_h - jpp.t_pad + jpp.kh - jpp.ih) in execute_backward()
928 - nstl::max(0, jpp.t_pad - oh * jpp.stride_h)); in execute_backward()
939 for (int oh = 0; oh < jpp.oh; ++oh) in execute_backward()
947 const auto nb2_c = utils::div_up(jpp.nb_c, jpp.ur_bc); in execute_backward()
949 = static_cast<std::size_t>(jpp.mb) * nb2_c; in execute_backward()
953 transpose_facade.fill_input_c_tail_with_zeros(ithr, jpp); in execute_backward()
958 utils::nd_iterator_init(start, n, jpp.mb, b2_c, nb2_c); in execute_backward()
960 const auto b_c = b2_c * jpp.ur_bc; in execute_backward()
961 const auto ur_bc = nstl::min(jpp.ur_bc, jpp.nb_c - b_c); in execute_backward()
964 utils::nd_iterator_step(n, jpp.mb, b2_c, nb2_c); in execute_backward()
979 const auto &jpp = pd()->jpp_; in execute_backward_3d() local
986 = bwd_pooling_transpose_facade_t<data_t, wsp_data_t, d_type>(jpp, in execute_backward_3d()
995 nstl::max(oh * jpp.stride_h - jpp.t_pad + jpp.kh, 0), jpp.ih); in execute_backward_3d()
1000 nstl::max(od * jpp.stride_d - jpp.f_pad + jpp.kd, 0), jpp.id); in execute_backward_3d()
1008 const int ij = oh * jpp.stride_h; in execute_backward_3d()
1009 const int i_t_overflow = nstl::max(0, jpp.t_pad - ij); in execute_backward_3d()
1011 = nstl::max(jpp.ih, ij + jpp.kh - jpp.t_pad) - jpp.ih; in execute_backward_3d()
1012 const int ih = nstl::max(ij - jpp.t_pad, 0); in execute_backward_3d()
1014 = ((jpp.tag_kind == jit_memory_tag_kind_t::nspc) ? jpp.c_block in execute_backward_3d()
1019 arg.src = transpose_facade.get_src_addr_3d(ithr, id + kd, ih, jpp); in execute_backward_3d()
1025 arg.dst = transpose_facade.get_dst_addr_3d(ithr, od, oh, jpp); in execute_backward_3d()
1033 ithr, od, oh, jpp); in execute_backward_3d()
1043 = (od == jpp.od - 1) ? jpp.id : get_last_id(od); in execute_backward_3d()
1049 = (oh == jpp.oh - 1) ? jpp.ih : get_last_ih(oh); in execute_backward_3d()
1054 ithr, zero_id_start, zero_ih_start, jpp); in execute_backward_3d()
1063 arg.kd_padding = jpp.kd - d_t_overflow - d_b_overflow; in execute_backward_3d()
1064 arg.kh_padding = jpp.kh - i_t_overflow - i_b_overflow; in execute_backward_3d()
1065 arg.kh_padding_shift = i_t_overflow * jpp.kw in execute_backward_3d()
1066 + d_t_overflow * jpp.kw * jpp.kh + kd * jpp.kw * jpp.kh; in execute_backward_3d()
1067 arg.kd_padding_shift = (i_t_overflow + i_b_overflow) * jpp.kw; in execute_backward_3d()
1068 arg.ker_area_h = (float)(jpp.kh in execute_backward_3d()
1070 oh * jpp.stride_h - jpp.t_pad + jpp.kh in execute_backward_3d()
1071 - jpp.ih) in execute_backward_3d()
1072 - nstl::max(0, jpp.t_pad - oh * jpp.stride_h)) in execute_backward_3d()
1073 * (jpp.kd in execute_backward_3d()
1075 od * jpp.stride_d - jpp.f_pad + jpp.kd - jpp.id) in execute_backward_3d()
1076 - nstl::max(0, jpp.f_pad - od * jpp.stride_d)); in execute_backward_3d()
1084 const int ik = od * jpp.stride_d; in execute_backward_3d()
1085 const int d_t_overflow = nstl::max(0, jpp.f_pad - ik); in execute_backward_3d()
1087 = nstl::max(jpp.id, ik + jpp.kd - jpp.f_pad) - jpp.id; in execute_backward_3d()
1088 const int id = nstl::max(ik - jpp.f_pad, 0); in execute_backward_3d()
1090 for (int oh = 0; oh < jpp.oh; ++oh) { in execute_backward_3d()
1096 if (jpp.simple_alg) { in execute_backward_3d()
1097 if (jpp.tag_kind == jit_memory_tag_kind_t::nspc) { in execute_backward_3d()
1098 const auto nb2_c = utils::div_up(jpp.nb_c, jpp.ur_bc); in execute_backward_3d()
1099 parallel_nd(jpp.mb, jpp.od, nb2_c, [&](int n, int od, int b2_c) { in execute_backward_3d()
1100 const auto b_c = b2_c * jpp.ur_bc; in execute_backward_3d()
1101 const auto ur_bc = nstl::min(jpp.ur_bc, jpp.nb_c - b_c); in execute_backward_3d()
1105 assert(jpp.ur_bc == 1); in execute_backward_3d()
1107 parallel_nd_ext(0, jpp.mb, jpp.nb_c, in execute_backward_3d()
1112 for (int od = 0; od < jpp.od; ++od) { in execute_backward_3d()
1121 jpp.mb, jpp.nb_c, jpp.od, [&](int n, int b_c, int od) { in execute_backward_3d()
1128 if (jpp.tag_kind == jit_memory_tag_kind_t::nspc) { in execute_backward_3d()
1129 const size_t chunk_size = (size_t)jpp.ih * jpp.iw * jpp.c; in execute_backward_3d()
1130 parallel_nd(jpp.mb, jpp.id, [&](int n, int id) { in execute_backward_3d()
1131 const size_t offset = ((size_t)n * jpp.id + id) * chunk_size; in execute_backward_3d()
1139 = (size_t)jpp.id * jpp.ih * jpp.iw * jpp.c_block; in execute_backward_3d()
1140 parallel_nd_ext(0, jpp.mb, jpp.nb_c, in execute_backward_3d()
1143 = ((size_t)n * jpp.nb_c + b_c) * chunk_size; in execute_backward_3d()
1151 const auto nb2_c = utils::div_up(jpp.nb_c, jpp.ur_bc); in execute_backward_3d()
1154 0, jpp.mb, nb2_c, [&](int ithr, int nthr, int n, int b2_c) { in execute_backward_3d()
1155 const auto b_c = b2_c * jpp.ur_bc; in execute_backward_3d()
1161 size_t block_size = jpp.c_block * jpp.id * jpp.ih in execute_backward_3d()
1162 * jpp.iw * jpp.dt_size; in execute_backward_3d()
1165 ithr, 0, 0, jpp); in execute_backward_3d()
1169 for (int kd = 0; kd < jpp.kd; ++kd) { in execute_backward_3d()
1171 = nstl::min(jpp.ur_bc, jpp.nb_c - b_c); in execute_backward_3d()
1172 for (int od = 0; od < jpp.od; ++od) { in execute_backward_3d()
1173 const int ik = od * jpp.stride_d; in execute_backward_3d()
1175 = nstl::max(0, jpp.f_pad - ik); in execute_backward_3d()
1177 = nstl::max(jpp.id, in execute_backward_3d()
1178 ik + jpp.kd - jpp.f_pad) in execute_backward_3d()
1179 - jpp.id; in execute_backward_3d()
1180 if (kd >= jpp.kd - d_t_overflow - d_b_overflow) in execute_backward_3d()
1182 const int id = nstl::max(ik - jpp.f_pad, 0); in execute_backward_3d()
1183 for (int oh = 0; oh < jpp.oh; ++oh) { in execute_backward_3d()
1196 for (int kd = 0; kd < jpp.kd; ++kd) { in execute_backward_3d()
1197 parallel_nd(jpp.mb, nb2_c, [&](int n, int b2_c) { in execute_backward_3d()
1198 const auto b_c = b2_c * jpp.ur_bc; in execute_backward_3d()
1199 const auto ur_bc = nstl::min(jpp.ur_bc, jpp.nb_c - b_c); in execute_backward_3d()
1200 for (int od = 0; od < jpp.od; ++od) { in execute_backward_3d()
1201 const int ik = od * jpp.stride_d; in execute_backward_3d()
1202 const int d_t_overflow = nstl::max(0, jpp.f_pad - ik); in execute_backward_3d()
1204 = nstl::max(jpp.id, ik + jpp.kd - jpp.f_pad) in execute_backward_3d()
1205 - jpp.id; in execute_backward_3d()
1206 if (kd >= jpp.kd - d_t_overflow - d_b_overflow) in execute_backward_3d()
1208 const int id = nstl::max(ik - jpp.f_pad, 0); in execute_backward_3d()
1209 for (int oh = 0; oh < jpp.oh; ++oh) { in execute_backward_3d()