Home
last modified time | relevance | path

Searched refs:OptimizeFrozenGraph (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/torch/csrc/jit/passes/
Dfrozen_graph_optimizations.h16 TORCH_API void OptimizeFrozenGraph(
Dfrozen_graph_optimizations.cpp13 void OptimizeFrozenGraph( in OptimizeFrozenGraph() function
/external/pytorch/torch/csrc/jit/backends/xnnpack/
Dxnnpack_graph_builder.cpp27 OptimizeFrozenGraph(graph, true); in optimizeAndTraceGraph()
/external/pytorch/torch/csrc/jit/api/
Dmodule.cpp488 OptimizeFrozenGraph(graph, optimize_numerics); in freeze()
/external/pytorch/torch/csrc/jit/python/
Dinit.cpp420 .def("_jit_pass_optimize_frozen_graph", &OptimizeFrozenGraph) in initJITBindings()