Searched refs:tgt_seq_length_max (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 | 54 const dim_t tgt_seq_length_max = 10; variable
183 std::vector<float> net_dst(batch * tgt_seq_length_max * feature_size, 1.0f); in simple_net()
540 tgt_seq_length_max * batch * feature_size, 1.0f); in simple_net()
726 for (dim_t i = 0; i < tgt_seq_length_max; i++) { in simple_net()
785 << " maximum target sequence length = " << tgt_seq_length_max in simple_net()
|
| H A D | cpu_rnn_inference_int8.cpp | 54 const dim_t tgt_seq_length_max = 10; variable
210 std::vector<float> net_dst(batch * tgt_seq_length_max * feature_size, 0.1f); in simple_net()
632 std::vector<uint8_t> dec_dst(tgt_seq_length_max * batch * feature_size, 0); in simple_net()
811 for (dim_t i = 0; i < tgt_seq_length_max; i++) { in simple_net()
872 << " maximum target sequence length = " << tgt_seq_length_max in simple_net()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/examples/ |
| H A D | cpu_rnn_inference_f32.cpp | 54 const dim_t tgt_seq_length_max = 10; variable
187 std::vector<float> net_dst(batch * tgt_seq_length_max * feature_size, 1.0f); in simple_net()
543 tgt_seq_length_max * batch * feature_size, 1.0f); in simple_net()
729 for (dim_t i = 0; i < tgt_seq_length_max; i++) { in simple_net()
788 << " maximum target sequence length = " << tgt_seq_length_max in simple_net()
|
| H A D | cpu_rnn_inference_int8.cpp | 54 const dim_t tgt_seq_length_max = 10; variable
228 std::vector<float> net_dst(batch * tgt_seq_length_max * feature_size, 0.1f); in simple_net()
637 std::vector<uint8_t> dec_dst(tgt_seq_length_max * batch * feature_size, 0); in simple_net()
820 for (dim_t i = 0; i < tgt_seq_length_max; i++) { in simple_net()
881 << " maximum target sequence length = " << tgt_seq_length_max in simple_net()
|
| /dports/math/onednn/oneDNN-2.5.1/examples/ |
| H A D | cpu_rnn_inference_f32.cpp | 54 const dim_t tgt_seq_length_max = 10; variable
187 std::vector<float> net_dst(batch * tgt_seq_length_max * feature_size, 1.0f); in simple_net()
543 tgt_seq_length_max * batch * feature_size, 1.0f); in simple_net()
729 for (dim_t i = 0; i < tgt_seq_length_max; i++) { in simple_net()
788 << " maximum target sequence length = " << tgt_seq_length_max in simple_net()
|
| H A D | cpu_rnn_inference_int8.cpp | 54 const dim_t tgt_seq_length_max = 10; variable
228 std::vector<float> net_dst(batch * tgt_seq_length_max * feature_size, 0.1f); in simple_net()
637 std::vector<uint8_t> dec_dst(tgt_seq_length_max * batch * feature_size, 0); in simple_net()
820 for (dim_t i = 0; i < tgt_seq_length_max; i++) { in simple_net()
881 << " maximum target sequence length = " << tgt_seq_length_max in simple_net()
|