Searched defs:rnn_output (Results 1 – 3 of 3) sorted by relevance
600 auto rnn_output = in TEST_F() local616 auto rnn_output = sequential->forward< in TEST_F() local632 auto rnn_output = in TEST_F() local648 auto rnn_output = sequential->forward<torch::Tensor>(x); in TEST_F() local657 auto rnn_output = in TEST_F() local668 auto rnn_output = sequential->forward<torch::Tensor>(x); in TEST_F() local
726 auto rnn_output = m->forward_with_packed_input(packed_input); in TEST_F() local744 auto rnn_output = m->forward_with_packed_input(packed_input); in TEST_F() local762 auto rnn_output = m->forward_with_packed_input(packed_input); in TEST_F() local
1146 const float* rnn_output = fw_output.data() + s * rnn.num_fw_units(); in TEST() local