Searched defs:InnerTestModule (Results 1 – 1 of 1) sorted by relevance
5233 struct InnerTestModule : torch::nn::Module { in TEST_F() struct5234 InnerTestModule() in TEST_F() argument5239 torch::nn::Linear fc; in TEST_F()5240 torch::nn::Embedding table; in TEST_F()