Lines Matching refs:reduce_work

45         int channel, int kh_padding, int reduce_work, int load_work) {  in jit_conv_ker_pipeline()  argument
54 PIPELINE(reduce_work); in jit_conv_ker_pipeline()
62 const void *bias, int channel, int kh_padding, int iwb, int reduce_work, in jit_conv_ker_pipeline_iw_thr() argument
67 reduce_work, load_work); in jit_conv_ker_pipeline_iw_thr()
73 int reduce_work, int load_work) { in jit_conv_3d_ker_pipeline() argument
83 PIPELINE(reduce_work); in jit_conv_3d_ker_pipeline()
91 int channel, int kh_padding, int owb, int reduce_work, int load_work, in jit_conv_ker_pipeline_ow_thr() argument
102 reduce_work, load_work); in jit_conv_ker_pipeline_ow_thr()
109 int reduce_work, int load_work, int flags) { in jit_conv_3d_ker_pipeline_ow_thr() argument
114 kd_padding, reduce_work, load_work); in jit_conv_3d_ker_pipeline_ow_thr()
122 int reduce_work, int load_work, const void *post_ops_binary_rhs_arg_vec, in jit_conv_3d_ker_pipeline_ow_thr() argument
130 kh_padding, kd_padding, owb, reduce_work, load_work, flags); in jit_conv_3d_ker_pipeline_ow_thr()
135 const void *bias, int channel, int kh_padding, size_t reduce_work, in jit_conv_ker_pipeline_bwd_w() argument
138 reduce_work, load_work); in jit_conv_ker_pipeline_bwd_w()
144 int kh_padding /* kh_work_size */, size_t kh_offset, size_t reduce_work, in jit_conv_2d_ker_bwd_w_pipeline() argument
157 PIPELINE(reduce_work); in jit_conv_2d_ker_bwd_w_pipeline()
166 int kd_padding /* kd_work_size */, size_t kd_offset, size_t reduce_work, in jit_conv_3d_ker_bwd_w_pipeline() argument
179 PIPELINE(reduce_work); in jit_conv_3d_ker_bwd_w_pipeline()
752 int reduce_work = ocb_step * jcp.oc_block; in execute_backward_data_1d() local
756 reduce_work = utils::this_block_size(ocb * jcp.oc_block, in execute_backward_data_1d()
762 reduce_work, load_work); in execute_backward_data_1d()
875 int reduce_work = ocb_step * jcp.oc_block; in execute_backward_data_2d() local
879 reduce_work = utils::this_block_size(ocb * jcp.oc_block, in execute_backward_data_2d()
932 k_len, iwb, reduce_work, load_work); in execute_backward_data_2d()
1088 int reduce_work = ocb_step * jcp.oc_block; in execute_backward_data_3d() local
1092 reduce_work = utils::this_block_size(ocb * jcp.oc_block, in execute_backward_data_3d()
1146 k_len, d_len, reduce_work, load_work); in execute_backward_data_3d()