Searched refs:SRC_ITER (Results 1 – 10 of 10) sorted by relevance
/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/ |
H A D | cfg.cpp | 74 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 D | rnn.cpp | 220 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 D | rnn_aux.cpp | 112 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 D | rnn.hpp | 53 SRC_ITER, enumerator
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/ |
H A D | cfg.cpp | 70 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 D | rnn.cpp | 220 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 D | rnn_aux.cpp | 112 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 D | rnn.hpp | 53 SRC_ITER, enumerator
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/nn/mkldnn/ |
H A D | mkldnn_rnn.cc | 351 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 D | mkldnn_rnn.cc | 351 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()
|