Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/cpp/api/
H A Drnn.cpp17 auto rnn_model = model_maker(nhid); in test_RNN_xor() local
18 auto rnn = model->add(rnn_model, "rnn"); in test_RNN_xor()
20 if (rnn_model.get()->options_base.proj_size() > 0) { in test_RNN_xor()
21 nout = rnn_model.get()->options_base.proj_size(); in test_RNN_xor()