| /external/pytorch/test/jit/ |
| D | test_torchbind.py | 113 scripted = torch.jit.script(foo)
114 script_output = scripted(script_input)
122 scripted = torch.jit.script(foo)
130 fc.run(str(scripted.graph))
131 out = scripted()
142 scripted = torch.jit.script(foo)
143 out = scripted()
175 scripted = torch.jit.script(foo_just_getter)
176 out, result = scripted()
194 scripted = torch.jit.script(foo_not_setter)
[all …]
|
| D | test_enum.py | 252 scripted = torch.jit.script(m)
256 ).check_next("prim::EnumValue").check_next("return").run(str(scripted.graph))
258 self.assertEqual(scripted(), Color.RED.value)
275 scripted = torch.jit.script(m)
277 self.assertEqual(scripted(), (Color.RED.name, Color.RED.value))
311 scripted = torch.jit.script(m)
315 ).check_next("return").run(str(scripted.graph))
317 self.assertEqual(scripted(), Color.RED)
333 scripted = torch.jit.script(iterate_enum)
337 ).check_same("Color.GREEN").check_same("Color.BLUE").run(str(scripted.graph))
[all …]
|
| D | test_union.py | 63 scripted = torch.jit.script(fn)
71 scripted("1")
80 scripted = torch.jit.script(fn)
89 scripted({"foo": "bar", "baz": "qux"})
98 scripted(["foo", "bar", "baz"])
107 scripted("1")
122 scripted = torch.jit.script(fn)
131 scripted(1)
145 scripted = torch.jit.script(fn)
153 scripted(["foo", "bar", "baz"])
[all …]
|
| D | test_union_pep604.py | 65 scripted = torch.jit.script(fn)
73 scripted("1")
82 scripted = torch.jit.script(fn)
91 scripted({"foo": "bar", "baz": "qux"})
100 scripted(["foo", "bar", "baz"])
109 scripted("1")
124 scripted = torch.jit.script(fn)
133 scripted(1)
147 scripted = torch.jit.script(fn)
155 scripted(["foo", "bar", "baz"])
[all …]
|
| D | test_tensor_methods.py | 33 scripted = torch.jit.script(tensor_getitem)
34 FileCheck().check("aten::index").run(scripted.graph)
|
| D | test_autodiff_subgraph_slicing.py | 170 scripted = self.checkScript(method1, (x, weight, b1, b2))
174 scripted,
191 scripted = self.checkScript(method1, (x, weight, bias))
195 scripted,
202 scripted = self.checkScript(method1, (x, weight, bias))
206 scripted,
|
| D | test_misc.py | 48 scripted = torch.jit.script(func)
458 scripted = torch.jit.script(foo)
459 actual = scripted(x, 0)
|
| D | test_types.py | 44 scripted = torch.jit.script(fn)(x)
46 self.assertEqual(expected, scripted)
|
| D | test_complex.py | 617 scripted = torch.jit.script(op)
618 jit_result = scripted(x, y)
|
| D | test_with.py | 628 scripted = torch.jit.script(with_rf)
631 scripted(x, y)
|
| D | test_recursive_script.py | 344 scripted = torch.jit.script(a)
379 scripted = torch.jit.script(M())
|
| /external/pytorch/torch/testing/_internal/ |
| D | static_module.py | 8 def __init__(self, scripted): argument
10 if hasattr(scripted, "_c"):
11 self.static_module = torch._C._jit_to_static_module(scripted._c)
13 self.static_module = torch._C._jit_to_static_module(scripted.graph)
|
| /external/pytorch/torch/csrc/jit/ |
| D | JIT-AUTOCAST.md | 20 …- [Disabling eager autocast with scripted autocast](#disabling-eager-autocast-with-scripted-autoca…
38 Explicit `with autocast()` scopes are supported inside scripted functions and
148 Calling a traced function from a scripted one mostly works, except for the case
172 Calling a scripted function from a trace is similar to calling the scripted
191 #### Disabling eager autocast with scripted autocast
194 within a scripted function, autocasting will still occur.
|
| /external/pytorch/test/mobile/lightweight_dispatch/ |
| D | tests_setup.py | 23 scripted = torch.jit.script(model)
24 buffer = BytesIO(scripted._save_to_buffer_for_lite_interpreter())
31 scripted._save_for_lite_interpreter(path)
|
| /external/pytorch/test/quantization/core/ |
| D | test_workflow_module.py | 265 scripted = torch.jit.script(obs)
269 scripted(x)
270 self.assertEqual(obs.calculate_qparams(), scripted.calculate_qparams())
273 torch.jit.save(scripted, buf)
397 scripted = torch.jit.script(model)
398 scripted = torch.ao.quantization.prepare_jit(scripted, qconfig_dict)
400 scripted(x)
401 obs_dict = torch.ao.quantization.get_observer_state_dict(scripted)
408 self.assertEqual(scripted.state_dict(), scripted_2.state_dict())
640 scripted = torch.jit.script(obs)
[all …]
|
| /external/pytorch/test/ |
| D | test_ops_jit.py | 332 scripted = torch.jit.CompilationUnit(script)._fn
337 scripted(inp)
345 scripted(inp)
347 graph = scripted.graph_for(inp)
|
| D | test_jit_fuser.py | 61 scripted = self.checkScript(func, (a,))
62 self.assertAllFused(scripted.graph_for(a))
186 scripted = self.checkScript(f, (x, y))
187 self.assertAllFused(scripted.graph_for(x, y))
569 scripted = self.checkScript(f, (x,))
570 self.assertAllFused(scripted.graph_for(x))
579 scripted = self.checkScript(fn_test_scalar_arg, (x, p))
580 self.assertAllFused(scripted.graph_for(x, p))
589 scripted = torch.jit.script(fn_test_scalar_arg_requires_grad)
590 out = scripted(x, p)
[all …]
|
| D | test_jit_fuser_te.py | 219 scripted = self.checkScript(fused_kernel, (a, a))
220 graph = scripted.graph_for(a, a)
231 self.assertEqual(scripted(a, a), fused_kernel(a, a))
241 scripted = self.checkScript(func, (a,))
257 scripted = self.checkScript(func, (a,))
259 scripted = self.checkScript(func_neg, (a,))
280 scripted = self.checkScript(func, (a,))
297 scripted = self.checkScript(foo, inputs)
298 self.assertAllFused(scripted.graph_for(*inputs))
401 scripted = self.checkScript(f, (x, y))
[all …]
|
| /external/pytorch/test/package/ |
| D | test_model.py | 188 scripted = torch.jit.script(loaded)
192 torch.jit.save(scripted, f2)
|
| D | test_package_script.py | 36 scripted = torch.jit.script(uses_interface)
37 scripted.proxy_mod = torch.jit.script(fake.NewModule())
53 self.assertEqual(scripted(input), scripted_loaded(input))
|
| /external/pytorch/torch/jit/ |
| D | supported_ops.py | 104 scripted = torch.jit.script(attr)
105 scripted_schema = scripted.schema
|
| D | _recursive.py | 601 scripted = interface_script(module_type, orig_value)
603 scripted = orig_value
606 scripted = create_script_module_impl(
610 cpp_module.setattr(name, scripted)
611 script_module._modules[name] = scripted
|
| /external/pytorch/test/quantization/bc/ |
| D | test_backward_compatibility.py | 160 scripted = torch.jit.script(qmodule)
170 scripted, qconfig_dict, _eval_fn, [input_tensor]
|
| /external/pytorch/docs/source/ |
| D | torch.compiler_custom_backends.rst | 253 scripted = torch.jit.script(gm)
254 return torch.jit.optimize_for_inference(scripted)
|
| /external/skia/bazel/rbe/ |
| D | README.md | 30 In accordance with SLSA level 1, we want to be able to have a scripted way of building our image
|