Searched refs:ow_mask (Results 1 – 1 of 1) sorted by relevance
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/conv/ |
| H A D | kernel_builder.cpp | 5952 expr_t od_mask, oh_mask, ow_mask; in init_fwd() local
6004 if (check_ow) ow_mask = (x >= 0) & (x < cfg_.ow); in init_fwd()
6067 dst_view.set_tdim(5, ow, ow_mask); in init_fwd()
6145 expr_t od_mask(true), oh_mask(true), ow_mask(true); in init_bwd_d() local
6179 if (check_ow) ow_mask = (x >= 0) & (x < cfg_.ow); in init_bwd_d()
6206 dst_view.set_tdim(4, ow / cfg_.sw, ow_mask & (ow % cfg_.sw == 0)); in init_bwd_d()
6301 expr_t od_mask, oh_mask, ow_mask; in init_bwd_w() local
6353 if (check_ow) ow_mask = (x < cfg_.ow); in init_bwd_w()
6404 dst_view.set_tdim(4, ow, ow_mask); in init_bwd_w()
|