Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/benchmarks/instruction_counts/definitions/
H A Dstandard.py219 torchscript=torchscript,
221 for setup, torchscript, (py_constructor, cpp_constructor) in (
252 torchscript=True,
271 torchscript=True,
/aosp_15_r20/external/pytorch/benchmarks/instruction_counts/core/
H A Dapi.py162 torchscript: bool
174 torchscript: bool = False,
196 if torchscript
209 torchscript=torchscript,
222 torchscript: bool = False,
251 torchscript=torchscript,
H A Dexpand.py219 if args.py_model_setup and args.torchscript:
237 if runtime == RuntimeMode.JIT and not args.torchscript:
/aosp_15_r20/external/pytorch/torch/csrc/lazy/generated/
H A DREADME.md1 This folder contains generated sources for the lazy torchscript backend.
3 The main input file that drives which operators get codegen support for torchscript backend is
/aosp_15_r20/external/pytorch/
H A DGLOSSARY.md19 - [TorchScript](#torchscript)
/aosp_15_r20/external/pytorch/torch/_dynamo/backends/
H A Ddebugging.py86 def torchscript(gm, fake_tensor_inputs): function
/aosp_15_r20/external/pytorch/benchmarks/instruction_counts/
H A DREADME.md54 torchscript=True,
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_utility_funs.py1776 def _test_deduplicate_initializers(self, torchscript=False): argument
1799 model = torch.jit.script(MyModule()) if torchscript else MyModule()
1838 self._test_deduplicate_initializers(torchscript=False)
1841 self._test_deduplicate_initializers(torchscript=True)
/aosp_15_r20/external/pytorch/android/
H A DREADME.md234 To load torchscript model for mobile we need some special setup which is placed in `struct JITCallG…
/aosp_15_r20/external/pytorch/docs/source/notes/
H A Dcpu_threading_torchscript_inference.rst1 .. _cpu-threading-torchscript-inference:
/aosp_15_r20/external/pytorch/docs/source/
H A Djit.rst761 Everything in a user defined `TorchScript Class <torchscript-class>`_ is
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dutils.py926 def torchscript(model, example_inputs, verbose=False): function
/aosp_15_r20/external/pytorch/torch/csrc/jit/
H A DOVERVIEW.md1202 device, and shape - they are only applied after the first invocation of a torchscript function or
1203 module. As a result, the first invocation of a torchscript function can sometimes behave slightly