Home
last modified time | relevance | path

Searched refs:checkScript (Results 1 – 25 of 32) sorted by relevance

12

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_complex.py25 self.checkScript(fn, (3 + 5j,))
32 self.checkScript(fn, (input, 2))
39 self.checkScript(fn, (input, -4.3 - 2j))
68 self.checkScript(fn, (t1, t2, 2))
150 self.checkScript(fn, (val,))
158 self.checkScript(pow_float_complex, (2, 3j))
159 self.checkScript(pow_complex_float, (3j, 2))
168 self.checkScript(pow_complex_complex, (x, y))
176 self.checkScript(
217 self.checkScript(
[all …]
H A Dtest_hash.py29 self.checkScript(fn, ((1, 2), (1, 2)))
30 self.checkScript(fn, ((1, 2), (3, 4)))
31 self.checkScript(fn, ((1, 2), (2, 1)))
53 self.checkScript(fn, (tensor1, tensor1))
54 self.checkScript(fn, (tensor1, tensor1_clone))
55 self.checkScript(fn, (tensor1, tensor2))
63 self.checkScript(fn, ())
69 self.checkScript(fn, (True, False))
70 self.checkScript(fn, (True, True))
71 self.checkScript(fn, (False, True))
[all …]
H A Dtest_list_dict.py45 self.checkScript(if_predicate, ([1, 2, 3],))
46 self.checkScript(if_predicate, ([],))
54 self.checkScript(while_predicate, ([1, 2, 3],))
55 self.checkScript(while_predicate, ([],))
60 self.checkScript(ternary_predicate, ([1, 2, 3],))
61 self.checkScript(ternary_predicate, ([],))
67 self.checkScript(int_in, ([1, 2, 3],))
68 self.checkScript(int_in, ([1, 3, 3],))
73 self.checkScript(float_in, ([1.0, 2.0, 3.0],))
74 self.checkScript(float_in, ([1.0, 3.0, 3.0],))
[all …]
H A Dtest_typing.py31 self.checkScript(test_in_dict, ({"hi": 2, "bye": 3},))
32 self.checkScript(test_in_dict, ({"bye": 3},))
62 self.checkScript(test_not_in_dict, ({"hello": 1, "world": 2},))
63 self.checkScript(test_not_in_dict, ({"world": 2},))
75 self.checkScript(test_dict_tensor_key, (dict_a, torch.tensor(4)))
76 self.checkScript(test_dict_tensor_key, (dict_a, torch.tensor(3)))
77 self.checkScript(test_dict_tensor_key, (dict_a, inp1))
78 self.checkScript(test_dict_tensor_key, (dict_a, inp2))
175 self.checkScript(stuff, (a,))
182 self.checkScript(bar, ())
[all …]
H A Dtest_union_pep604.py54 self.checkScript(test_func, (5, 6))
62 self.checkScript(fn, (1,))
63 self.checkScript(fn, (1.0,))
79 self.checkScript(fn, ({"foo": 1, "bar": 2, "baz": 3},))
80 self.checkScript(fn, ([1, 2, 3],))
121 self.checkScript(fn, (Color.RED,))
122 self.checkScript(fn, ("red",))
161 self.checkScript(fn, (1,))
168 self.checkScript(fn, ())
178 self.checkScript(fn, ())
[all …]
H A Dtest_union.py52 self.checkScript(test_func, (5, 6))
60 self.checkScript(fn, (1,))
61 self.checkScript(fn, (1.0,))
77 self.checkScript(fn, ({"foo": 1, "bar": 2, "baz": 3},))
78 self.checkScript(fn, ([1, 2, 3],))
119 self.checkScript(fn, (Color.RED,))
120 self.checkScript(fn, ("red",))
159 self.checkScript(fn, (1,))
166 self.checkScript(fn, ())
176 self.checkScript(fn, ())
[all …]
H A Dtest_isinstance.py33 self.checkScript(int_test, (x,))
41 self.checkScript(float_test, (x,))
49 self.checkScript(bool_test, (x,))
58 self.checkScript(list_str_test, (x,))
66 self.checkScript(list_tensor_test, (x,))
75 self.checkScript(dict_str_int_test, (x,))
82 self.checkScript(dict_int_tensor_test, (x,))
91 self.checkScript(tuple_test, (x,))
98 self.checkScript(tuple_tensor_test, (x,))
106 self.checkScript(optional_test, (x,))
[all …]
H A Dtest_slice.py39 self.checkScript(three_nones, (range(10),))
45 self.checkScript(two_nones, (range(10),))
51 self.checkScript(one_none, (range(10),))
57 self.checkScript(fn, (range(10),))
63 self.checkScript(fn, (range(10),))
69 self.checkScript(fn, (range(10),))
75 self.checkScript(fn, (range(10),))
81 self.checkScript(fn, (range(10),))
87 self.checkScript(fn, ("foo_bar",))
93 self.checkScript(fn, (torch.ones(10),))
[all …]
H A Dtest_string_formatting.py29 self.checkScript(fn, (5, 2))
35 self.checkScript(fn, ("foo",))
43 self.checkScript(fn, ("%s in template",))
49 self.checkScript(fn, ("foo",))
65 self.checkScript(fn, (1,))
71 self.checkScript(fn, (1,))
77 self.checkScript(fn, (1,))
109 self.checkScript(fn, (1,))
117 self.checkScript(fn, (1,))
123 self.checkScript(fn, (1.0,))
[all …]
H A Dtest_python_builtins.py44 self.checkScript(func, (a, b), optimize=True)
52 self.checkScript(func, (a, b), optimize=True)
70 self.checkScript(fn, (a, b), optimize=True)
94 self.checkScript(func, (a, b), optimize=True)
95 self.checkScript(func2, (a, b, c, d), optimize=True)
96 self.checkScript(func3, (4, -0.5), optimize=True)
97 self.checkScript(func4, ())
110 self.checkScript(func5, (x, y))
117 self.checkScript(func, [x], optimize=True)
124 self.checkScript(func, [x], optimize=True)
[all …]
H A Dtest_with.py177 self.checkScript(test_basic, (test_input,))
178 self.checkScript(test_pass, (test_input,))
179 self.checkScript(test_early_return, (test_input, test_context))
180 self.checkScript(test_break, (test_input, test_context, test_list))
181 self.checkScript(test_continue, (test_input, test_context, test_list))
183 self.checkScript(test_serial, (test_input,))
184 self.checkScript(test_nested, (test_input,))
185 self.checkScript(test_combined, (test_input,))
334 self.checkScript(test_basic, (test_input,))
335 self.checkScript(test_pass, (test_input,))
[all …]
H A Dtest_tensor_creation_ops.py35 self.checkScript(randperm, (3,))
44 self.checkScript(randperm, (3,))
53 self.checkScript(triu_indices, (3, 3))
62 self.checkScript(triu_indices, (3, 3))
71 self.checkScript(tril_indices, (3, 3))
80 self.checkScript(tril_indices, (3, 3))
H A Dtest_class_type.py385 sfoo = self.checkScript(use_foo, input)
415 self.checkScript(test, ([Foo(2), Foo(1), Foo(3)],))
416 self.checkScript(test, ([Foo(2), Foo(1), Foo(3)], True))
417 self.checkScript(test, ([Foo(2)],))
418 self.checkScript(test, ([],))
660 self.checkScript(use_them, (torch.rand(3, 4),))
767 self.checkScript(test_overload, ())
907 self.checkScript(func, ())
1229 self.checkScript(all_defaults, ())
1230 self.checkScript(some_defaults, ())
[all …]
H A Dtest_builtins.py90 self.checkScript(fn, ([1, 2, 3],))
123 self.checkScript(fn, ([1, 2, 3],))
217 self.checkScript(fn, (torch.zeros(2, dtype=torch.uint8),))
226 self.checkScript(fn6, (torch.zeros(2, dtype=torch.float32, device="cuda"),))
235 self.checkScript(test_scalar_cast, (torch.tensor(1.0),))
236 self.checkScript(test_scalar_cast, (torch.tensor(1),))
H A Dtest_peephole.py28 self.checkScript(test_write, (torch.ones(4, 4),))
47 j = self.checkScript(test_write, (a,))
55 j = self.checkScript(test_peephole, (a,))
170 self.checkScript(is_int_tensor, (torch.tensor(2),))
171 self.checkScript(is_int_tensor, (torch.tensor(2.5),))
241 self.checkScript(convertible_rsub, (a, b))
250 self.checkScript(convertible_is_op, (True, False))
260 self.checkScript(convertible_isnot_op, (True, False))
829 self.checkScript(foo, (3,))
836 self.checkScript(fn, (3,))
[all …]
H A Dtest_dataclasses.py91 self.checkScript(fn, (1.0, 2.0, 3))
103 self.checkScript(fn, [x, y])
126 self.checkScript(compare, [x1, y1, x2, y2])
159 self.checkScript(fn, [1, 2, 3])
H A Dtest_torchbind.py158 self.checkScript(foo_getter_setter_full, ())
167 self.checkScript(foo_getter_setter_lambda, ())
203 self.checkScript(foo_readwrite, ())
445 self.checkScript(fn, (1,))
459 self.checkScript(fn, ())
472 self.checkScript(gn, ())
H A Dtest_misc.py171 self.checkScript(if_function, (torch.randn(5),))
312 self.checkScript(foo, ())
322 self.checkScript(foo, ())
358 self.checkScript(test_all_dtypes, ())
373 self.checkScript(no_inputs, ())
H A Dtest_remove_mutation.py258 self.checkScript(successful, ())
269 self.checkScript(successful, ())
279 self.checkScript(successful, ())
H A Dtest_types.py40 self.checkScript(fn, [torch.randn(2, 2)])
52 self.checkScript(fn, [torch.randn(2, 2)])
139 self.checkScript(fn, ())
H A Dtest_autodiff_subgraph_slicing.py170 scripted = self.checkScript(method1, (x, weight, b1, b2))
191 scripted = self.checkScript(method1, (x, weight, bias))
202 scripted = self.checkScript(method1, (x, weight, bias))
/aosp_15_r20/external/pytorch/test/
H A Dtest_jit_string.py68 self.checkScript(test_str_is, (input,))
69 self.checkScript(test_str_to, (input,))
70 self.checkScript(test_str_strip, (input,))
72 self.checkScript(test_str_strip_char_set, (input, char_set))
74 self.checkScript(test_str_center, (i, input,))
75 self.checkScript(test_str_center_fc, (i, input,))
76 self.checkScript(test_ljust, (input, i))
77 self.checkScript(test_ljust_fc, (input, i, '*'))
78 self.checkScript(test_rjust, (input, i))
79 self.checkScript(test_rjust_fc, (input, i, '*'))
[all …]
H A Dtest_jit.py863 self.checkScript(T, (x, ))
864 self.checkScript(mT, (x, ))
865 self.checkScript(H, (x, ))
866 self.checkScript(mH, (x, ))
867 self.checkScript(T, (y, ))
868 self.checkScript(mT, (y, ))
869 self.checkScript(H, (y, ))
870 self.checkScript(mH, (y, ))
908 jitted = self.checkScript(foo, (x,))
1142 self.checkScript(test_simple_grad, (x, y), inputs_requires_grad=True)
[all …]
H A Dtest_jit_fuser.py61 scripted = self.checkScript(func, (a,))
97 ge = self.checkScript(decode, inputs)
186 scripted = self.checkScript(f, (x, y))
197 ge = self.checkScript(cuda_rem, inputs)
209 ge = self.checkScript(fn, inputs)
242 self.checkScript(fn, [tensor])
283 module = self.checkScript(func, inputs)
306 ge = self.checkScript(fn, inputs)
324 s = self.checkScript(f, inputs)
349 s = self.checkScript(f, (inp1, inp2), profiling=ProfilingMode.PROFILING)
[all …]
H A Dtest_jit_fuser_te.py219 scripted = self.checkScript(fused_kernel, (a, a))
241 scripted = self.checkScript(func, (a,))
257 scripted = self.checkScript(func, (a,))
259 scripted = self.checkScript(func_neg, (a,))
270 self.checkScript(func, (a,))
280 scripted = self.checkScript(func, (a,))
297 scripted = self.checkScript(foo, inputs)
310 ge = self.checkScript(decode, inputs)
342 self.checkScript(scaleshift, inputs)
401 scripted = self.checkScript(f, (x, y))
[all …]

12