Searched refs:compute_attention (Results 1 – 6 of 6) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/mkldnn_compat/examples/ |
| H A D | cpu_rnn_inference_f32.cpp | 82 void compute_attention(float *context_vectors, dim_t src_seq_length_max, in compute_attention() function
737 compute_attention(src_att_iter_handle, src_seq_length_max, batch, in simple_net()
|
| H A D | cpu_rnn_inference_int8.cpp | 94 void compute_attention(float *context_vectors, dim_t src_seq_length_max, in compute_attention() function
822 compute_attention(src_att_iter_handle, src_seq_length_max, batch, in simple_net()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/examples/ |
| H A D | cpu_rnn_inference_f32.cpp | 82 void compute_attention(float *context_vectors, dim_t src_seq_length_max, in compute_attention() function
740 compute_attention(src_att_iter_handle, src_seq_length_max, batch, in simple_net()
|
| H A D | cpu_rnn_inference_int8.cpp | 94 void compute_attention(float *context_vectors, dim_t src_seq_length_max, in compute_attention() function
831 compute_attention(src_att_iter_handle, src_seq_length_max, batch, in simple_net()
|
| /dports/math/onednn/oneDNN-2.5.1/examples/ |
| H A D | cpu_rnn_inference_f32.cpp | 82 void compute_attention(float *context_vectors, dim_t src_seq_length_max, in compute_attention() function
740 compute_attention(src_att_iter_handle, src_seq_length_max, batch, in simple_net()
|
| H A D | cpu_rnn_inference_int8.cpp | 94 void compute_attention(float *context_vectors, dim_t src_seq_length_max, in compute_attention() function
831 compute_attention(src_att_iter_handle, src_seq_length_max, batch, in simple_net()
|