Searched defs:input_cuda (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | rnn.cpp | 581 auto input_cuda = torch::tensor({1, 2, 3, 4, 5, 6}, input_opt) in TEST_F() local
635 auto input_cuda = torch::tensor({1, 2, 3, 4, 5, 6}, options) in TEST_F() local
692 auto input_cuda = torch::tensor({1, 2, 3, 4, 5, 6}, options) in TEST_F() local
|