Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_torchbind.py53 val = torch.classes._TorchScriptTesting._Foo(5, 3)
60 val = torch.classes._TorchScriptTesting._Foo(5, 3)
64 ss = torch.classes._TorchScriptTesting._StackString(["asdf", "bruh"])
70 ss1 = torch.classes._TorchScriptTesting._StackString(["asdf", "bruh"])
71 ss2 = torch.classes._TorchScriptTesting._StackString(["111", "222"])
97 foo = torch.classes._TorchScriptTesting._Foo(int1, int2)
105 StackString = torch.classes._TorchScriptTesting._StackString
112 script_input = torch.classes._TorchScriptTesting._StackString([])
119 ss = torch.classes._TorchScriptTesting._StackString(["hi", "mom"])
137 ss = torch.classes._TorchScriptTesting._StackString(["hi", "mom"])
[all …]
/aosp_15_r20/external/pytorch/test/export/
H A Dtest_torchbind.py111 torch.ops._TorchScriptTesting.takes_foo,
112 torch.ops._TorchScriptTesting.takes_foo_python_meta,
113 torch.ops._TorchScriptTesting.takes_foo_list_return,
114 torch.ops._TorchScriptTesting.takes_foo_tuple_return,
115 torch.ops._TorchScriptTesting.take_an_instance,
116 torch.ops._TorchScriptTesting.take_an_instance_inferred,
117 torch.ops._TorchScriptTesting.takes_foo_cia,
118 torch.ops._TorchScriptTesting.queue_pop,
119 torch.ops._TorchScriptTesting.queue_push,
120 torch.ops._TorchScriptTesting.queue_size,
[all …]
H A Dtest_passes.py94 self.attr = torch.classes._TorchScriptTesting._Foo(10, 20)
99 self.attr = torch.classes._TorchScriptTesting._Foo(10, 20)
101 torch.classes._TorchScriptTesting._Foo(1, 2),
103 torch.classes._TorchScriptTesting._Foo(3, 4),
105 {"foo": torch.classes._TorchScriptTesting._Foo(5, 6)},
111 self.attr = torch.classes._TorchScriptTesting._Foo(10, 20)
113 torch.classes._TorchScriptTesting._Foo(1, 2),
115 torch.classes._TorchScriptTesting._Foo(3, 4),
117 {"foo": torch.classes._TorchScriptTesting._Foo(5, 6)},
125 self.attr = torch.classes._TorchScriptTesting._Foo(10, 20)
[all …]
H A Dtest_lift_unlift.py173 builder.constant("const_obj", torch.classes._TorchScriptTesting._Foo(10, 20))
176 torch.ops._TorchScriptTesting.takes_foo,
294 const_obj = torch.classes._TorchScriptTesting._Foo(10, 20)
306 torch.ops._TorchScriptTesting.takes_foo,
311 torch.ops._TorchScriptTesting.takes_foo,
396 const_obj = torch.classes._TorchScriptTesting._Foo(10, 20)
H A Dtest_serialize.py979 self.attr = torch.classes._TorchScriptTesting._Foo(10, 20)
982 a = torch.ops._TorchScriptTesting.takes_foo_tuple_return(self.attr, x)
984 b = torch.ops._TorchScriptTesting.takes_foo(self.attr, y)
995 self.attr = torch.classes._TorchScriptTesting._Foo(10, 20)
998 a = torch.ops._TorchScriptTesting.takes_foo(self.attr, x)
999 b = torch.ops._TorchScriptTesting.takes_foo(self.attr, a)
1010 self.attr = torch.classes._TorchScriptTesting._Foo(10, 20)
1013 a = torch.ops._TorchScriptTesting.takes_foo_list_return(self.attr, x)
1015 b = torch.ops._TorchScriptTesting.takes_foo(self.attr, y)
1197 custom_obj = torch.classes._TorchScriptTesting._PickleTester([3, 4])
[all …]
H A Dtest_converter.py54 torch.ops._TorchScriptTesting.queue_pop,
55 torch.ops._TorchScriptTesting.queue_push,
56 torch.ops._TorchScriptTesting.queue_size,
1321 torch.ops._TorchScriptTesting.queue_push(self.tq, x.cos())
1322 return torch.ops._TorchScriptTesting.queue_pop(self.tq), self.tq.pop()
H A Dtest_unflatten.py625 self.attr = torch.classes._TorchScriptTesting._Foo(10, 20)
H A Dtest_export.py7965 custom_obj = torch.classes._TorchScriptTesting._PickleTester([3, 4])
7985 torch.ops._TorchScriptTesting.take_an_instance.default,
8012 == torch.ops._TorchScriptTesting.take_an_instance.default
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dtorchbind_impls.py27 return torch.classes._TorchScriptTesting._TensorQueue(
63 torch.ops._TorchScriptTesting.takes_foo_list_return.default.py_impl(
67 torch.ops._TorchScriptTesting.takes_foo_tuple_return.default.py_impl(
71 torch.ops._TorchScriptTesting.takes_foo.default.py_impl(torch._C.DispatchKey.Meta)(
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_torchbind.py26 self.attr = torch.classes._TorchScriptTesting._Foo(10, 20)
31 a = torch.ops._TorchScriptTesting.takes_foo_tuple_return(self.attr, x)
33 b = torch.ops._TorchScriptTesting.takes_foo(self.attr, y)
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_custom_class_registrations.cpp428 TORCH_LIBRARY(_TorchScriptTesting, m) { in TORCH_LIBRARY() argument
690 TORCH_LIBRARY_FRAGMENT(_TorchScriptTesting, m) { in TORCH_LIBRARY_FRAGMENT() argument
702 TORCH_LIBRARY_IMPL(_TorchScriptTesting, CPU, m) { in TORCH_LIBRARY_IMPL() argument
711 TORCH_LIBRARY_IMPL(_TorchScriptTesting, Meta, m) { in TORCH_LIBRARY_IMPL() argument
717 TORCH_LIBRARY_IMPL(_TorchScriptTesting, CompositeImplicitAutograd, m) { in TORCH_LIBRARY_IMPL() argument
723 TORCH_LIBRARY_IMPL(_TorchScriptTesting, BackendSelect, m) { in TORCH_LIBRARY_IMPL() argument
/aosp_15_r20/external/pytorch/test/
H A Dtest_weak.py572 torch.classes._TorchScriptTesting._Foo(1, 2): 1,
573 torch.classes._TorchScriptTesting._Foo(2, 3): 2,
574 torch.classes._TorchScriptTesting._Foo(3, 4): 3,
H A Dtest_fx.py816 interpreter = torch.classes._TorchScriptTesting._ElementwiseInterpreter()
2650 self.f = torch.classes._TorchScriptTesting._StackString(["3", "4"])
2665 self.f = torch.classes._TorchScriptTesting._ReLUClass()
/aosp_15_r20/external/pytorch/test/higher_order_ops/
H A Dtest_with_effects.py144 self.attr = torch.classes._TorchScriptTesting._Foo(10, 20)
147 return (x + torch.ops._TorchScriptTesting.takes_foo(self.attr, x),)
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_control_flow.py3978 foo = torch.classes._TorchScriptTesting._Foo(3, 4)