Home
last modified time | relevance | path

Searched refs:lstm_output_to_device (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp/api/
H A Drnn.cpp428 lstm_output_to_device( in lstm_output_to_device() function
524 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()