Home
last modified time | relevance | path

Searched defs:block_k (Results 1 – 22 of 22) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/gemm/
H A Dgen9_gemm.cpp33 static constexpr int block_k = 1024; member
39 static constexpr int block_k = 2048; member
45 static constexpr int block_k = 2048; member
52 int block_m, block_n, block_k; in get_blocking() local
337 int block_m, block_n, block_k; in execute_standard() local
455 auto block_k = driver_params_f32_nocopy_t::block_k; in execute_superkernel() local
H A Dxe_lp_gemm.cpp31 static constexpr auto block_k = 1024; member
223 int block_m, block_n, block_k; in execute_standard() local
H A Dgen9_gemm_x8x8s32.cpp32 static constexpr auto block_k member
219 int block_m, block_n, block_k; in execute_standard() local
H A Dgen9_gemm.hpp191 int block_m, block_n, block_k; in init_scratchpad_copy_based() local
/dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/gemm/
H A Dgen9_gemm.cpp33 static constexpr int block_k = 1024; member
39 static constexpr int block_k = 2048; member
45 static constexpr int block_k = 2048; member
52 int block_m, block_n, block_k; in get_blocking() local
337 int block_m, block_n, block_k; in execute_standard() local
455 auto block_k = driver_params_f32_nocopy_t::block_k; in execute_superkernel() local
H A Dgen9_gemm_x8x8s32.cpp32 static constexpr auto block_k member
219 int block_m, block_n, block_k; in execute_standard() local
H A Dxe_lp_gemm.cpp31 static constexpr auto block_k = 1024; member
226 int block_k = xe_lp_gemm_driver_params_t::block_k; in execute_standard() local
H A Dgen9_gemm.hpp207 int block_m, block_n, block_k; in init_scratchpad_copy_based() local
/dports/multimedia/recordmydesktop/recordmydesktop-0.3.8.1/src/
H A Drmd_cache_frame.c58 block_k=(!blockwidth)?0:(blockno%(width/blockwidth)); in FlushBlock() local
H A Drmd_load_cache.c77 block_k=blockno%(width/blockwidth); in LoadBlock() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/gemm/
H A Dgemm_threading.hpp47 dim_t block_m, block_n, block_k; // Blocking sizes (-1 = default) member
H A Dgemm_driver.cpp169 dim_t block_m, block_k; in get_k_padd() local
173 dim_t block_n, block_k; in get_k_padd() local
195 dim_t block_m, block_k; in get_m_padd() local
222 dim_t block_n, block_k; in get_n_padd() local
1187 auto &block_k = thread_info.block_k; in set_thread_opts_pack() local
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/gemm/
H A Dgemm_threading.hpp47 dim_t block_m, block_n, block_k; // Blocking sizes (-1 = default) member
H A Dgemm_driver.cpp169 dim_t block_m, block_k; in get_k_padd() local
173 dim_t block_n, block_k; in get_k_padd() local
195 dim_t block_m, block_k; in get_m_padd() local
222 dim_t block_n, block_k; in get_n_padd() local
1187 auto &block_k = thread_info.block_k; in set_thread_opts_pack() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/jit/gemm/
H A Dgen_gemm.cpp192 auto block_k in execute() local
/dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/gemm/
H A Dgen_gemm.cpp223 auto block_k = nocopy_info_.blocking[2]; in execute() local
H A Dxe_hp_systolic_gemm.cpp616 int64_t block_k = compute_info_.blocking[LoopK]; in get_blocking() local
871 int64_t block_m = 0, block_n = 0, block_k = 0; in execute() local
/dports/math/viennacl/ViennaCL-1.7.1/viennacl/linalg/detail/spai/
H A Dfspai.hpp181 MatrixT & block_k = blocks[k]; in fill_blocks() local
/dports/misc/tvm/incubator-tvm-0.6.1/tutorials/optimize/
H A Dopt_conv_tensorcore.py267 block_k = s[Conv].fuse(hc, wc) variable
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tutorials/optimize/
H A Dopt_conv_tensorcore.py330 block_k = s[Conv].fuse(hc, wc) variable
/dports/misc/py-tvm/incubator-tvm-0.6.1/tutorials/optimize/
H A Dopt_conv_tensorcore.py267 block_k = s[Conv].fuse(hc, wc) variable
/dports/science/py-scipy/scipy-1.7.1/scipy/sparse/sparsetools/
H A Dbsr.h52 const npy_intp block_k = brow * R + k - bcol * C; in bsr_diagonal() local