Home
last modified time | relevance | path

Searched refs:optimize_numerics (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/jit/
H A D_freeze.py15 mod, preserved_attrs: Optional[List[str]] = None, optimize_numerics: bool = True
121 run_frozen_optimizations(out, optimize_numerics, preserved_methods)
127 mod, optimize_numerics: bool = True, preserved_methods: Optional[List[str]] = None
171 torch._C._jit_pass_optimize_frozen_graph(mod.graph, optimize_numerics)
178 mod.__getattr__(method).graph, optimize_numerics
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dfrozen_graph_optimizations.cpp15 bool optimize_numerics) { in OptimizeFrozenGraph() argument
19 if (optimize_numerics) { in OptimizeFrozenGraph()
H A Dfrozen_graph_optimizations.h18 bool optimize_numerics = true);
/aosp_15_r20/external/pytorch/torch/csrc/jit/api/
H A Dmodule.cpp480 bool optimize_numerics) { in freeze() argument
488 OptimizeFrozenGraph(graph, optimize_numerics); in freeze()
H A Dmodule.h340 bool optimize_numerics = true);
/aosp_15_r20/external/pytorch/torch/_C/
H A D__init__.pyi.in305 def _jit_pass_optimize_frozen_graph(Graph, optimize_numerics: _bool = True) -> None: ...
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_freezing.py2694 torch.jit.script(mod.eval()), optimize_numerics=False