Searched refs:optimize_numerics (Results 1 – 7 of 7) sorted by relevance
/aosp_15_r20/external/pytorch/torch/jit/ |
H A D | _freeze.py | 15 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 D | frozen_graph_optimizations.cpp | 15 bool optimize_numerics) { in OptimizeFrozenGraph() argument 19 if (optimize_numerics) { in OptimizeFrozenGraph()
|
H A D | frozen_graph_optimizations.h | 18 bool optimize_numerics = true);
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/api/ |
H A D | module.cpp | 480 bool optimize_numerics) { in freeze() argument 488 OptimizeFrozenGraph(graph, optimize_numerics); in freeze()
|
H A D | module.h | 340 bool optimize_numerics = true);
|
/aosp_15_r20/external/pytorch/torch/_C/ |
H A D | __init__.pyi.in | 305 def _jit_pass_optimize_frozen_graph(Graph, optimize_numerics: _bool = True) -> None: ...
|
/aosp_15_r20/external/pytorch/test/jit/ |
H A D | test_freezing.py | 2694 torch.jit.script(mod.eval()), optimize_numerics=False
|