Home
last modified time | relevance | path

Searched refs:insertPrePackedOps (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/csrc/jit/passes/
Dxnnpack_rewrite.cpp373 void insertPrePackedOps(std::shared_ptr<Graph>& graph) { in insertPrePackedOps() function
378 void insertPrePackedOps(script::Module& module) { in insertPrePackedOps() function
381 insertPrePackedOps(graph); in insertPrePackedOps()
384 insertPrePackedOps(m); in insertPrePackedOps()
441 insertPrePackedOps(cloned_module); in optimizeForMobile()
482 void insertPrePackedOps(std::shared_ptr<Graph>& graph) { in insertPrePackedOps() function
487 void insertPrePackedOps(script::Module& module) { in insertPrePackedOps() function
Dxnnpack_rewrite.h11 TORCH_API void insertPrePackedOps(std::shared_ptr<Graph>& graph);
12 TORCH_API void insertPrePackedOps(script::Module& module);
/external/pytorch/torch/csrc/jit/python/
Dinit.cpp1051 return insertPrePackedOps(graph); in initJITBindings()
1055 [](script::Module& module) { return insertPrePackedOps(module); }) in initJITBindings()