Searched refs:lstm_cpu (Results 1 – 1 of 1) sorted by relevance
613 LSTM lstm_cpu{opt}; in TEST_F() local619 for (const auto& param : lstm_cpu->named_parameters(/*recurse=*/false)) { in TEST_F()621 lstm_cpu->named_parameters()[param.key()]); in TEST_F()625 lstm_cpu->flatten_parameters(); in TEST_F()640 auto output_cpu = lstm_cpu->forward(input_cpu); in TEST_F()670 LSTM lstm_cpu{opt}; in TEST_F() local676 for (const auto& param : lstm_cpu->named_parameters(/*recurse=*/false)) { in TEST_F()678 lstm_cpu->named_parameters()[param.key()]); in TEST_F()682 lstm_cpu->flatten_parameters(); in TEST_F()697 auto output_cpu = lstm_cpu->forward(input_cpu); in TEST_F()