Searched refs:reinterpret_as_acc (Results 1 – 2 of 2) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/rnn/ |
| H A D | ref_postgemm_gru.cpp | 41 T4 src_to_float, T5 reinterpret_as_acc, const float *scales, in gru_fwd_part1_postgemm_template() argument
70 scratch_gates(i, 0, j) = reinterpret_as_acc(G0); in gru_fwd_part1_postgemm_template()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/rnn/ |
| H A D | ref_postgemm_gru.cpp | 41 T4 src_to_float, T5 reinterpret_as_acc, const float *scales, in gru_fwd_part1_postgemm_template() argument
76 scratch_gates(i, 0, j) = reinterpret_as_acc(G0); in gru_fwd_part1_postgemm_template()
|