Home
last modified time | relevance | path

Searched defs:U_sz (Results 1 – 8 of 8) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/
H A Dgen9_wino_convolution.cpp168 size_t U_sz = conf.tile_size * conf.kh * conf.wino_ic * conf.wino_oc; in init_conf() local
279 size_t U_sz = conf.tile_size * conf.kh * conf.wino_ic * conf.wino_oc; in init_scratchpad() local
/dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/
H A Dgen9_wino_convolution.cpp168 size_t U_sz = conf.tile_size * conf.kh * conf.wino_ic * conf.wino_oc; in init_conf() local
279 size_t U_sz = conf.tile_size * conf.kh * conf.wino_ic * conf.wino_oc; in init_scratchpad() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Djit_avx512_common_convolution_winograd.hpp40 size_t U_sz = (size_t)alpha * alpha * jcp.ic * jcp.oc; in init_scratchpad() local
H A Djit_avx512_core_f32_wino_conv_4x3.hpp40 size_t U_sz = (size_t)alpha * alpha * jcp.ic * jcp.oc; in init_scratchpad() local
H A Djit_avx512_core_f32_wino_conv_4x3.cpp661 const int U_sz = nthreads * alpha * alpha * jcp.oc / jcp.nb_oc * jcp.ic in _execute_backward_weights_SDGtWo() local
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_avx512_core_f32_wino_conv_4x3.hpp40 size_t U_sz = (size_t)alpha * alpha * jcp.ic * jcp.oc; in init_scratchpad() local
H A Djit_avx512_common_convolution_winograd.hpp40 size_t U_sz = (size_t)alpha * alpha * jcp.ic * jcp.oc; in init_scratchpad() local
H A Djit_avx512_core_f32_wino_conv_4x3.cpp661 const int U_sz = nthreads * alpha * alpha * jcp.oc / jcp.nb_oc * jcp.ic in _execute_backward_weights_SDGtWo() local