Home
last modified time | relevance | path

Searched refs:MyModule (Results 1 – 25 of 88) sorted by relevance

1234

/external/pytorch/torch/testing/_internal/distributed/rpc/
Ddist_optimizer_test.py17 class MyModule: class
104 remote_module1 = rpc.remote(owner1, MyModule)
105 remote_module2 = rpc.remote(owner2, MyModule)
106 remote_param1 = remote_method(MyModule.get_w, remote_module1)
107 remote_param2 = remote_method(MyModule.get_w, remote_module2)
118 output1 = rpc_async_method(MyModule.forward, remote_module1, t2)
119 output2 = rpc_async_method(MyModule.forward, remote_module2, output1.wait())
132 remote_module1 = rpc.remote(owner1, MyModule)
133 remote_module2 = rpc.remote(owner2, MyModule)
134 remote_param1 = remote_method(MyModule.get_w, remote_module1)
[all …]
/external/pytorch/test/onnx/
Dtest_onnx_opset.py77 class MyModule(Module): class
84 check_onnx_opsets_operator(MyModule(), x, ops, opset_versions=[9, 10])
87 class MyModule(Module): class
106 check_onnx_opsets_operator(MyModule(), x, ops, opset_versions=[9, 10])
174 class MyModule(Module): class
180 module = MyModule()
202 class MyModule(Module): class
206 module = MyModule()
218 class MyModule(Module): class
241 check_onnx_opsets_operator(MyModule(), x, ops, opset_versions=[9, 10])
[all …]
Dtest_custom_ops.py28 class MyModule(torch.nn.Module): class
38 model = MyModule()
54 class MyModule(torch.nn.Module): class
81 model = MyModule()
/external/pytorch/benchmarks/dynamo/
Ddist_util.py49 class MyModule(torch.nn.Module): class
67 + [MyModule(10000, 10000)]
68 + [MyModule(10000, 1000)]
69 + [MyModule(1000, 1000)]
70 + [MyModule(1000, 1000)]
71 + [MyModule(1000, 1000)]
72 + [MyModule(1000, 1000)]
73 + [MyModule(1000, 1000)]
74 + [MyModule(1000, 1000)]
75 + [MyModule(1000, 1000)]
[all …]
/external/dagger2/javatests/dagger/functional/kotlinsrc/nullables/
DJspecifyNullableTest.kt31 @Component(modules = [MyIntComponent.MyModule::class])
36 class MyModule(val intValue: Int) { in getInt() class
42 @Component(modules = [MyNullableStringComponent.MyModule::class])
47 class MyModule(val stringValue: String?) { in getString() class
57 .myModule(MyIntComponent.MyModule(15)) in testIntWithValue()
66 .myModule(MyNullableStringComponent.MyModule("TEST_VALUE")) in testStringWithValue()
75 .myModule(MyNullableStringComponent.MyModule(null)) in testStringWithNull()
/external/jackson-databind/src/test/java/com/fasterxml/jackson/databind/module/
DTestDuplicateRegistration.java8 static class MyModule extends com.fasterxml.jackson.databind.Module { class in TestDuplicateRegistration
11 public MyModule() { in MyModule() method in TestDuplicateRegistration.MyModule
36 MyModule module = new MyModule(); in testDuplicateRegistration()
47 final MyModule module2 = new MyModule(); in testDuplicateRegistration()
/external/jackson-databind/src/test/java/com/fasterxml/jackson/databind/deser/creators/
DTestValueInstantiator.java288 static class MyModule extends SimpleModule class in TestValueInstantiator
290 public MyModule(Class<?> cls, ValueInstantiator inst) in MyModule() method in TestValueInstantiator.MyModule
343 mapper.registerModule(new MyModule(MyBean.class, new MyBeanInstantiator())); in testCustomBeanInstantiator()
352 mapper.registerModule(new MyModule(MyList.class, new MyListInstantiator())); in testCustomListInstantiator()
362 mapper.registerModule(new MyModule(MyMap.class, new MyMapInstantiator())); in testCustomMapInstantiator()
378 mapper.registerModule(new MyModule(MyBean.class, new MyDelegateBeanInstantiator())); in testDelegateBeanInstantiator()
387 mapper.registerModule(new MyModule(MyList.class, new MyDelegateListInstantiator())); in testDelegateListInstantiator()
397 mapper.registerModule(new MyModule(MyMap.class, new MyDelegateMapInstantiator())); in testDelegateMapInstantiator()
422 mapper.registerModule(new MyModule(CreatorBean.class, in testPropertyBasedBeanInstantiator()
449 mapper.registerModule(new MyModule(MyMap.class, new CreatorMapInstantiator())); in testPropertyBasedMapInstantiator()
[all …]
/external/dagger2/javatests/dagger/functional/nullables/
DJspecifyNullableTest.java33 @Component(modules = MyModule.class, dependencies = ComponentDependency.class)
45 static class MyModule { class in JspecifyNullableTest
49 MyModule(Integer integer, InnerType innerType) { in MyModule() method in JspecifyNullableTest.MyModule
87 .myModule(new MyModule(15, new InnerType() {})) in testWithValue()
100 .myModule(new MyModule(null, null)) in testWithNull()
/external/dagger2/javatests/dagger/internal/codegen/goldens/
DModuleFactoryGeneratorTest_testScopedMetadataOnNonStaticProvides_test.MyModule_ProvideStringFactory27 private final MyModule module;
29 public MyModule_ProvideStringFactory(MyModule module) {
38 public static MyModule_ProvideStringFactory create(MyModule module) {
42 public static String provideString(MyModule instance) {
/external/pytorch/test/inductor/
Dtest_standalone_compile.py11 class MyModule(torch.nn.Module): class
24 class MyModule2(MyModule):
30 class MyModule3(MyModule):
50 mod = MyModule().eval()
66 mod = MyModule().eval()
/external/executorch/exir/backend/test/
Dtest_utils.py34 class MyModule(torch.nn.Module): class
42 m = MyModule()
48 class MyModule(torch.nn.Module): class
55 m = MyModule()
66 class MyModule(torch.nn.Module): class
75 graph_module: torch.fx.GraphModule = symbolic_trace(MyModule())
/external/pytorch/test/jit/
Dtest_module_containers.py278 class MyModule(torch.nn.Module): class
327 m = MyModule()
345 class MyModule(torch.nn.Module): class
356 m = MyModule()
361 class MyModule(torch.nn.Module): class
375 m = MyModule()
440 class MyModule(torch.nn.Module): class
453 m = MyModule()
712 class MyModule(nn.Module): class
725 self.checkModule(MyModule(), (torch.zeros(1)))
[all …]
Dtest_save_load_for_op_version.py85 class MyModule(torch.nn.Module): class
102 current_mobile_module = self._save_load_mobile_module(MyModule)
137 class MyModule(torch.nn.Module): class
150 current_mobile_module = self._save_load_mobile_module(MyModule)
191 class MyModule(torch.nn.Module): class
203 current_mobile_module = self._save_load_mobile_module(MyModule)
427 class MyModule(torch.nn.Module): class
443 current_mobile_module = self._save_load_mobile_module(MyModule)
/external/pytorch/test/export/
Dtest_unflatten.py81 class MyModule(torch.nn.Module): class
96 orig_eager = MyModule()
124 class MyModule(torch.nn.Module): class
136 eager_module = MyModule()
159 class MyModule(torch.nn.Module): class
172 eager_module = MyModule()
231 class MyModule(torch.nn.Module): class
242 orig_eager = MyModule()
391 class MyModule(torch.nn.Module): class
406 orig_eager = MyModule()
[all …]
Dtest_serialize.py148 class MyModule(torch.nn.Module): class
161 model = MyModule().eval()
180 class MyModule(torch.nn.Module): class
200 ep = export(MyModule(), inp)
211 class MyModule(torch.nn.Module): class
223 ep = export(MyModule(), inp)
233 class MyModule(torch.nn.Module): class
247 MyModule(),
303 class MyModule(torch.nn.Module): class
311 exported_module = export(MyModule(), (input,)).run_decompositions()
[all …]
Dtest_torchbind.py164 class MyModule(torch.nn.Module): class
173 MyModule(),
214 class MyModule(torch.nn.Module): class
223 MyModule(), (torch.ones(2, 3),), strict=False, pre_dispatch=pre_dispatch
248 class MyModule(torch.nn.Module): class
257 MyModule(), (torch.ones(2, 3),), strict=False, pre_dispatch=pre_dispatch
284 class MyModule(torch.nn.Module): class
292 MyModule(), (torch.ones(2, 3), cc), strict=False, pre_dispatch=pre_dispatch
322 class MyModule(torch.nn.Module): class
337 MyModule(),
[all …]
/external/pytorch/test/nn/
Dtest_load_state_dict.py266 class MyModule(torch.nn.Module): class
277 net = MyModule()
283 net_meta = MyModule()
342 class MyModule(torch.nn.Module): class
351 net = MyModule()
366 net_meta = MyModule()
392 class MyModule(torch.nn.Module): class
401 net = MyModule()
405 net2 = MyModule()
428 class MyModule(torch.nn.Module): class
[all …]
Dtest_module_hooks.py629 class MyModule(nn.Module): class
679 m = ctor(MyModule())
701 class MyModule(nn.Module): class
713 nested = MyModule()
816 class MyModule(torch.nn.Module): class
826 mod = MyModule()
1383 class MyModule(nn.Module): class
1391 mod = MyModule()
1450 class MyModule(nn.Module): class
1455 mod = MyModule()
[all …]
/external/guice/core/test/com/google/inject/example/
DClientServiceWithGuice.java44 public static class MyModule extends AbstractModule { class in ClientServiceWithGuice
88 Injector injector = Guice.createInjector(new MyModule()); in main()
/external/executorch/exir/tests/
Dtest_common.py54 class MyModule(torch.nn.Module): class
63 module = MyModule()
/external/pytorch/test/
Dtest_functional_optim.py15 class MyModule(torch.nn.Module): class
92 module_optim = MyModule()
93 module_functional = MyModule()
Dtest_subclass.py133 class MyModule(nn.Module): class
164 m = MyModule()
182 class MyModule(nn.Module): class
194 m = MyModule()
/external/pytorch/test/distributed/nn/jit/
Dtest_instantiator.py29 class MyModule(nn.Module): class
34 return MyModule()
/external/pytorch/test/dynamo/
Dtest_export.py204 class MyModule(torch.nn.Module): class
215 module = MyModule("moo")
1081 class MyModule(torch.nn.Module): class
1090 exported = torch._dynamo.export(MyModule(), aten_graph=False)(inp)
1101 exported = torch._dynamo.export(MyModule(), aten_graph=True)(inp)
1124 class MyModule(torch.nn.Module): class
1133 m = MyModule()
1194 class MyModule(torch.nn.Module): class
1210 module = MyModule()
1212 module = MyModule()
[all …]
/external/guice/core/test/com/google/inject/
DErrorHandlingTest.java33 Guice.createInjector(new MyModule()); in main()
124 static class MyModule extends AbstractModule { class in ErrorHandlingTest

1234