Searched refs:DIFF_WEIGHTS_ITER (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_aux.cpp | 127 CASE(DIFF_WEIGHTS_ITER); in data_kind2str() 540 case DIFF_WEIGHTS_ITER: in print_value()
|
H A D | rnn.cpp | 699 weights_iter_dims, prb.cfg[DIFF_WEIGHTS_ITER].dt, in init_pd() 1086 SAFE(fill_weights(prb, DIFF_WEIGHTS_ITER, diff_weights_iter_dt, in doit() 1159 DIFF_WEIGHTS_ITER, diff_weights_iter, tag::abx /*ldigo*/); in doit()
|
H A D | rnn.hpp | 66 DIFF_WEIGHTS_ITER, enumerator
|
/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/ |
H A D | rnn_aux.cpp | 127 CASE(DIFF_WEIGHTS_ITER); in data_kind2str() 540 case DIFF_WEIGHTS_ITER: in print_value()
|
H A D | rnn.cpp | 699 weights_iter_dims, prb.cfg[DIFF_WEIGHTS_ITER].dt, in init_pd() 1093 SAFE(fill_weights(prb, DIFF_WEIGHTS_ITER, diff_weights_iter_dt, in doit() 1166 DIFF_WEIGHTS_ITER, diff_weights_iter, tag::abx /*ldigo*/); in doit()
|
H A D | rnn.hpp | 66 DIFF_WEIGHTS_ITER, enumerator
|