Home
last modified time | relevance | path

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

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/nvidia/
H A Dcudnn_pooling.hpp102 ok = ok && blocking_ok(); in init()
114 bool blocking_ok() const { in blocking_ok() function
H A Dcudnn_conv_inner_product.hpp81 && blocking_ok(with_eltwise(), use_fused_path_for_blocking) in init()
168 bool blocking_ok( in blocking_ok() function
/dports/math/onednn/oneDNN-2.5.1/src/gpu/nvidia/
H A Dcudnn_pooling.hpp102 ok = ok && blocking_ok(); in init()
114 bool blocking_ok() const { in blocking_ok() function
H A Dcudnn_conv_inner_product.hpp81 && blocking_ok(with_eltwise(), use_fused_path_for_blocking) in init()
168 bool blocking_ok( in blocking_ok() function
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Djit_avx512_core_f32_wino_conv_4x3_kernel.cpp2325 auto blocking_ok = [&]() { in set_wsched_WEI_SDGtWo() local
2359 if (!(jcp.dimM_block % M_blk) && blocking_ok()) { in set_wsched_WEI_SDGtWo()
2403 auto blocking_ok = [&]() -> bool { in set_wsched_WEI_S_D_Giot_W() local
2436 if (blocking_ok()) { in set_wsched_WEI_S_D_Giot_W()
H A Djit_avx512_core_bf16_conv_kernel.cpp4360 const bool blocking_ok = large_diff_dst_size in init_conf() local
4368 bool use_spatial_blocking = jcp.is_1stconv && !nt_stores_ok && blocking_ok; in init_conf()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_avx512_core_f32_wino_conv_4x3_kernel.cpp2323 auto blocking_ok = [&]() { in set_wsched_WEI_SDGtWo() local
2357 if (!(jcp.dimM_block % M_blk) && blocking_ok()) { in set_wsched_WEI_SDGtWo()
2400 auto blocking_ok = [&]() -> bool { in set_wsched_WEI_S_D_Giot_W() local
2433 if (blocking_ok()) { in set_wsched_WEI_S_D_Giot_W()
H A Djit_brgemm_conv_utils.cpp1773 const status_t blocking_ok = cur_brgb.calc_blocks(); in init_conf() local
1774 if (blocking_ok != status::success) continue; in init_conf()
H A Djit_avx512_core_bf16_conv_kernel.cpp4351 const bool blocking_ok = large_diff_dst_size in init_conf() local
4359 bool use_spatial_blocking = jcp.is_1stconv && !nt_stores_ok && blocking_ok; in init_conf()