Lines Matching defs:M
139 struct M : torch::nn::Module { in TEST_F() struct
140 torch::Tensor forward(torch::Tensor input) { in TEST_F()
165 struct M : torch::nn::Cloneable<M> { in TEST_F() struct
166 void reset() override {} in TEST_F()
167 torch::Tensor forward(torch::Tensor input) { in TEST_F()
180 struct M : torch::nn::Cloneable<M> { in TEST_F() struct
181 void reset() override {} in TEST_F()
182 torch::Tensor forward(torch::Tensor input) { in TEST_F()
213 struct M : torch::nn::Cloneable<M> { in TEST_F() struct
214 void reset() override {} in TEST_F()
215 torch::Tensor forward(torch::Tensor input) { in TEST_F()
232 struct M : torch::nn::Cloneable<M> { in TEST_F() struct
233 M() { in TEST_F() argument
237 void reset() override { in TEST_F()
244 torch::Tensor forward(torch::Tensor x) { in TEST_F()
252 torch::nn::Conv2d conv{nullptr}; in TEST_F()
253 torch::nn::Linear fc{nullptr}; in TEST_F()