Home
last modified time | relevance | path

Searched defs:mean_mem (Results 1 – 13 of 13) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/
H A Dsimple_layer_normalization.hpp107 auto mean_mem = scratchpad.get_memory_storage(key_lnorm_tmp_mean); in execute() local
223 auto mean_mem = scratchpad.get_memory_storage(key_lnorm_tmp_mean); in execute() local
/dports/math/onednn/oneDNN-2.5.1/src/cpu/
H A Dsimple_layer_normalization.hpp107 auto mean_mem = scratchpad.get_memory_storage(key_lnorm_tmp_mean); in execute() local
224 auto mean_mem = scratchpad.get_memory_storage(key_lnorm_tmp_mean); in execute() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/examples/primitives/
H A Dlayer_normalization.cpp115 auto mean_mem = memory(lnorm_pd.mean_desc(), engine); in layer_normalization_example() local
H A Dbatch_normalization.cpp119 auto mean_mem = memory(bnorm_pd.mean_desc(), engine); in batch_normalization_example() local
/dports/math/onednn/oneDNN-2.5.1/examples/primitives/
H A Dlayer_normalization.cpp115 auto mean_mem = memory(lnorm_pd.mean_desc(), engine); in layer_normalization_example() local
H A Dbatch_normalization.cpp119 auto mean_mem = memory(bnorm_pd.mean_desc(), engine); in batch_normalization_example() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/examples/
H A Dbnorm_u8_via_binary_postops.cpp111 auto mean_mem = memory(mean_md, engine); in bnorm_u8_via_binary_postops() local
/dports/math/onednn/oneDNN-2.5.1/examples/
H A Dbnorm_u8_via_binary_postops.cpp111 auto mean_mem = memory(mean_md, engine); in bnorm_u8_via_binary_postops() local
/dports/graphics/blender/blender-2.91.0/intern/cycles/device/opencl/
H A Ddevice_opencl_impl.cpp1699 cl_mem mean_mem = CL_MEM_PTR(mean_ptr); in denoising_combine_halves() local
1756 cl_mem mean_mem = CL_MEM_PTR(mean_ptr); in denoising_get_feature() local
/dports/math/armadillo/armadillo-10.7.1/include/armadillo_bits/
H A Dgmm_full_meat.hpp1111 const eT* mean_mem = means.colptr(g); in internal_scalar_log_p() local
2555 eT* mean_mem = access::rw(means).colptr(g); in em_update_params() local
H A Dgmm_diag_meat.hpp2497 eT* mean_mem = access::rw(means).colptr(g); in em_update_params() local
/dports/math/R-cran-RcppArmadillo/RcppArmadillo/inst/include/armadillo_bits/
H A Dgmm_full_meat.hpp1111 const eT* mean_mem = means.colptr(g); in internal_scalar_log_p() local
2555 eT* mean_mem = access::rw(means).colptr(g); in em_update_params() local
H A Dgmm_diag_meat.hpp2497 eT* mean_mem = access::rw(means).colptr(g); in em_update_params() local