Searched defs:smodule (Results 1 – 5 of 5) sorted by relevance
27 auto smodule = StaticModule(m, /* is_frozen */ false, opts, {input1, input2}); in TEST() local61 auto smodule = StaticModule( in TEST() local100 auto smodule = StaticModule( in TEST() local
223 Node* getNodeWithKind(const StaticModule& smodule, const std::string& kind) { in getNodeWithKind()238 bool hasNodeWithKind(const StaticModule& smodule, const std::string& kind) { in hasNodeWithKind()310 auto smodule = test_context->makeStaticModule(opts); in testStaticRuntime() local406 torch::jit::StaticModule& smodule, in hasProcessedNodeWithName()
2601 torch::jit::StaticModule smodule(graph); in TEST() local2622 torch::jit::StaticModule smodule(graph); in TEST() local2667 torch::jit::StaticModule smodule(graph); in TEST() local2689 torch::jit::StaticModule smodule(graph); in TEST() local2782 torch::jit::StaticModule smodule(module); in TEST() local2815 torch::jit::StaticModule smodule(module); in TEST() local2831 torch::jit::StaticModule smodule(module); in TEST() local2860 torch::jit::StaticModule smodule(module); in TEST() local3207 torch::jit::StaticModule smodule(g); in TEST() local
930 torch::jit::StaticModule smodule(module); in TEST() local950 torch::jit::StaticModule smodule(module); in TEST() local976 torch::jit::StaticModule smodule(g); in TEST() local1093 torch::jit::StaticModule smodule(module); in TEST() local
1060 std::shared_ptr<StaticModule> smodule, in ForkedSubgraphSRLauncher()1127 auto smodule = in __anon69571ad16602() local