Home
last modified time | relevance | path

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

/dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/conv/
H A Dkernel_builder.cpp5985 int dst_oc_inner_blk = ir_utils::max_pow2_divisor(dst_oc); in init_fwd() local
5986 dst_oc_inner_blk = std::min(dst_oc_inner_blk, cfg_.oc_thr_blk); in init_fwd()
6014 dst_oc_mask = (x / dst_oc_inner_blk < dst_oc / dst_oc_inner_blk); in init_fwd()
6331 int dst_oc_inner_blk = ir_utils::max_pow2_divisor(dst_oc); in init_bwd_w() local
6335 dst_oc_inner_blk = std::min(dst_oc_inner_blk, cfg_.oc_thr_blk); in init_bwd_w()
6357 dst_oc_mask = (x / dst_oc_inner_blk < dst_oc / dst_oc_inner_blk); in init_bwd_w()