Searched refs:lstm_output_to_device (Results 1 – 1 of 1) sorted by relevance
428 lstm_output_to_device( in lstm_output_to_device() function524 bi_output = lstm_output_to_device(bi_output, torch::kCPU); in BidirectionalLSTMReverseForwardTest()525 reverse_output = lstm_output_to_device(reverse_output, torch::kCPU); in BidirectionalLSTMReverseForwardTest()643 output_cpu = lstm_output_to_device(output_cpu, torch::kCPU); in TEST_F()700 output_cpu = lstm_output_to_device(output_cpu, torch::kCPU); in TEST_F()