Home
last modified time | relevance | path

Searched refs:FloatListWrapperModule (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_native_functions.py10 class FloatListWrapperModule(torch.nn.Module): class
86 self.do_test_optional_floatlist_with_module(FloatListWrapperModule())
87 self.do_test_optional_floatlist_with_module(torch.jit.script(FloatListWrapperModule()))
105 FloatListWrapperModule()(torch.zeros(1), ["hi"])
108 torch.jit.script(FloatListWrapperModule())(torch.zeros(1), ["hi"])
111 FloatListWrapperModule()(torch.zeros(1), torch.zeros(1))
114 torch.jit.script(FloatListWrapperModule())(torch.zeros(1), torch.zeros(1))