/aosp_15_r20/external/pytorch/test/jit/ |
H A D | test_complex.py | 25 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 D | test_hash.py | 29 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 D | test_list_dict.py | 45 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 D | test_typing.py | 31 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 D | test_union_pep604.py | 54 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 D | test_union.py | 52 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 D | test_isinstance.py | 33 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 D | test_slice.py | 39 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 D | test_string_formatting.py | 29 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 D | test_python_builtins.py | 44 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 D | test_with.py | 177 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 D | test_tensor_creation_ops.py | 35 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 D | test_class_type.py | 385 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 D | test_builtins.py | 90 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 D | test_peephole.py | 28 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 D | test_dataclasses.py | 91 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 D | test_torchbind.py | 158 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 D | test_misc.py | 171 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 D | test_remove_mutation.py | 258 self.checkScript(successful, ()) 269 self.checkScript(successful, ()) 279 self.checkScript(successful, ())
|
H A D | test_types.py | 40 self.checkScript(fn, [torch.randn(2, 2)]) 52 self.checkScript(fn, [torch.randn(2, 2)]) 139 self.checkScript(fn, ())
|
H A D | test_autodiff_subgraph_slicing.py | 170 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 D | test_jit_string.py | 68 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 D | test_jit.py | 863 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 D | test_jit_fuser.py | 61 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 D | test_jit_fuser_te.py | 219 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 …]
|