Searched refs:wei_proj_nscales (Results 1 – 6 of 6) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/ |
| H A D | rnn.hpp | 306 wei_proj_nscales = dic; in prb_t()
310 wei_proj_nscales = 1; in prb_t()
315 = (float *)zmalloc(sizeof(float) * wei_proj_nscales, 64); in prb_t()
331 return wei_proj_scales[MIN2(idx, wei_proj_nscales - 1)]; in get_wei_proj_scale()
397 int wei_proj_nscales; member
|
| H A D | rnn.cpp | 113 *dnnl_attr, prb.wei_proj_nscales, prb.wei_proj_scales_mask, in create_dnnl_rnn_attr()
216 quantize(prb.wei_proj_scales, prb.wei_proj_nscales, 0, idx); in check_s8s8_reorder()
304 prb.wei_proj_scales, prb.wei_proj_nscales, 0.0f, idx); in fill_memory()
357 prb.wei_proj_scales, prb.wei_proj_nscales, idx); in fill_memory()
495 auto n_scales = (kind == WEIGHTS_PROJECTION) ? prb.wei_proj_nscales in fill_weights()
|
| H A D | rnn_aux.cpp | 752 if (with_projection) set_wei_scales(wei_proj_scales, wei_proj_nscales); in set_qparams()
|
| /dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/ |
| H A D | rnn.hpp | 306 wei_proj_nscales = dic; in prb_t()
310 wei_proj_nscales = 1; in prb_t()
315 = (float *)zmalloc(sizeof(float) * wei_proj_nscales, 64); in prb_t()
331 return wei_proj_scales[MIN2(idx, wei_proj_nscales - 1)]; in get_wei_proj_scale()
397 int wei_proj_nscales; member
|
| H A D | rnn.cpp | 113 *dnnl_attr, prb.wei_proj_nscales, prb.wei_proj_scales_mask, in create_dnnl_rnn_attr()
216 quantize(prb.wei_proj_scales, prb.wei_proj_nscales, 0, idx); in check_s8s8_reorder()
304 prb.wei_proj_scales, prb.wei_proj_nscales, 0.0f, idx); in fill_memory()
357 prb.wei_proj_scales, prb.wei_proj_nscales, idx); in fill_memory()
495 auto n_scales = (kind == WEIGHTS_PROJECTION) ? prb.wei_proj_nscales in fill_weights()
|
| H A D | rnn_aux.cpp | 752 if (with_projection) set_wei_scales(wei_proj_scales, wei_proj_nscales); in set_qparams()
|