Home
last modified time | relevance | path

Searched refs:SRC_ITER (Results 1 – 10 of 10) sorted by relevance

/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/
H A Dcfg.cpp74 CASE(SRC_ITER, BF16_ENTRY_BF16); in CFG()
85 CASE(SRC_ITER, BF16_ENTRY_BF16); in CFG()
144 CASE(SRC_ITER, U8_ENTRY_U8); in CFG()
160 CASE(SRC_ITER, U8_ENTRY_U8); in CFG()
176 CASE(SRC_ITER, U8_ENTRY_F32); in CFG()
192 CASE(SRC_ITER, U8_ENTRY_F32); in CFG()
208 CASE(SRC_ITER, S8_ENTRY_S8); in CFG()
224 CASE(SRC_ITER, S8_ENTRY_S8); in CFG()
240 CASE(SRC_ITER, S8_ENTRY_F32); in CFG()
256 CASE(SRC_ITER, S8_ENTRY_F32); in CFG()
H A Drnn.cpp220 case SRC_ITER: in check_s8s8_reorder()
293 assert(kind == SRC_LAYER || kind == SRC_ITER); in fill_memory()
314 case SRC_ITER: in fill_memory()
388 && (kind == SRC_LAYER || kind == SRC_ITER); in fill_activation()
446 + (1.f / prb.n_gates()) * prb.cfg[SRC_ITER].f_mean in fill_src_iter_c()
600 prb.cfg[SRC_ITER].dt, dnnl_ldnc), in init_pd()
945 SAFE(fill_activation(prb, SRC_ITER, src_iter_dt, src_iter_fp, rnn_attr), in doit()
H A Drnn_aux.cpp112 CASE(SRC_ITER); in data_kind2str()
454 if (kind == SRC_ITER || kind == DIFF_SRC_ITER) C = prb.sic; in inv_ldnc_off_f()
521 case SRC_ITER: in print_value()
H A Drnn.hpp53 SRC_ITER, enumerator
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/
H A Dcfg.cpp70 CASE(SRC_ITER, BF16_ENTRY_BF16); in CFG()
122 CASE(SRC_ITER, U8_ENTRY_U8); in CFG()
137 CASE(SRC_ITER, U8_ENTRY_U8); in CFG()
152 CASE(SRC_ITER, U8_ENTRY_F32); in CFG()
167 CASE(SRC_ITER, U8_ENTRY_F32); in CFG()
182 CASE(SRC_ITER, S8_ENTRY_S8); in CFG()
197 CASE(SRC_ITER, S8_ENTRY_S8); in CFG()
212 CASE(SRC_ITER, S8_ENTRY_F32); in CFG()
227 CASE(SRC_ITER, S8_ENTRY_F32); in CFG()
H A Drnn.cpp220 case SRC_ITER: in check_s8s8_reorder()
293 assert(kind == SRC_LAYER || kind == SRC_ITER); in fill_memory()
314 case SRC_ITER: in fill_memory()
388 && (kind == SRC_LAYER || kind == SRC_ITER); in fill_activation()
446 + (1.f / prb.n_gates()) * prb.cfg[SRC_ITER].f_mean in fill_src_iter_c()
600 prb.cfg[SRC_ITER].dt, dnnl_ldnc), in init_pd()
938 SAFE(fill_activation(prb, SRC_ITER, src_iter_dt, src_iter_fp, rnn_attr), in doit()
H A Drnn_aux.cpp112 CASE(SRC_ITER); in data_kind2str()
454 if (kind == SRC_ITER || kind == DIFF_SRC_ITER) C = prb.sic; in inv_ldnc_off_f()
521 case SRC_ITER: in print_value()
H A Drnn.hpp53 SRC_ITER, enumerator
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/nn/mkldnn/
H A Dmkldnn_rnn.cc351 RNN_FWD_SET(SRC_ITER, param_.state_dims, format_tag::ldnc, hx, dtype); in SetNewDataMem()
808 RNN_BWD_SET(SRC_ITER, fwd_args, diff_state); in SetDataGradsMem()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/nn/mkldnn/
H A Dmkldnn_rnn.cc351 RNN_FWD_SET(SRC_ITER, param_.state_dims, format_tag::ldnc, hx, dtype); in SetNewDataMem()
808 RNN_BWD_SET(SRC_ITER, fwd_args, diff_state); in SetDataGradsMem()