Home
last modified time | relevance | path

Searched refs:wei_g_mask (Results 1 – 1 of 1) sorted by relevance

/dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/conv/
H A Dkernel_builder.cpp5955 expr_t src_g_mask, wei_g_mask, dst_g_mask; in init_fwd() local
6008 wei_g_mask = (x / wei_g_inner_blk < wei_g / wei_g_inner_blk); in init_fwd()
6046 wei_view.set_tdim(0, g, wei_g_mask); in init_fwd()