Searched defs:input_reversed (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | rnn.cpp | 450 auto input_reversed = torch::tensor({5, 4, 3, 2, 1}, opt).reshape({5, 1, 1}); in BidirectionalGRUReverseForward() local
504 auto input_reversed = torch::tensor({5, 4, 3, 2, 1}, opt).reshape({5, 1, 1}); in BidirectionalLSTMReverseForwardTest() local
|