Lines Matching +defs:test +defs:module
13 namespace test { namespace
23 Linear module(3, 4); in TEST_F() local
34 Linear module(3, 4); in TEST_F() local
61 TestModule module; in TEST_F() local
203 Linear module(3, 4); in TEST_F() local
235 Linear module(LinearOptions(10, 20).bias(false)); in test_DeviceOrDtypeConversionSkipsUndefinedTensor() local
251 BatchNorm1d module( in test_DeviceOrDtypeConversionSkipsUndefinedTensor() local
279 Linear module(LinearOptions(10, 20).bias(false)); in TEST_F() local
290 BatchNorm1d module( in TEST_F() local
299 BatchNorm1d module( in TEST_F() local
321 Linear module(128, 64); in TEST_F() local
353 Linear module(128, 64); in TEST_F() local
377 UnCloneable module; in TEST_F() local
389 Cloneable module; in TEST_F() local
443 auto module = std::make_shared<TestDistinctParametersModule>(); in TEST_F() local
450 auto module = std::make_shared<TestDistinctParametersModule>(); in TEST_F() local
459 auto module = std::make_shared<TestDistinctParametersModule>(); in TEST_F() local
492 auto module = std::make_shared<TestModule>(); in TEST_F() local
535 std::shared_ptr<TestModule> module; in TEST_F() member
635 ParameterTestModule module; in TEST_F() local
641 ParameterTestModule module; in TEST_F() local
659 BufferTestModule module; in TEST_F() local
665 BufferTestModule module; in TEST_F() local
803 TestModule module(1); in TEST_F() local
811 TestModule module(1); in TEST_F() local
822 TestModule module(1); in TEST_F() local
830 TestModule module(1); in TEST_F() local
852 int64_t get_test_container_item(std::shared_ptr<torch::nn::Module> module) { in get_test_container_item()
937 model->apply([&index](torch::nn::Module& module) { in TEST_F()
947 model->apply([&index](const torch::nn::Module& module) { in TEST_F()
958 [&index, expected](const std::string& name, torch::nn::Module& module) { in TEST_F()
975 const std::string& name, const torch::nn::Module& module) { in TEST_F()
988 model->apply([&index](const std::shared_ptr<torch::nn::Module>& module) { in TEST_F()
1001 const std::shared_ptr<torch::nn::Module>& module) { in TEST_F()
1011 TestModule module(1); in TEST_F() local
1018 TestModule module(1); in TEST_F() local
1023 auto module = std::make_shared<TestModule>(1); in TEST_F() local