pytorch/torch/csrc/jit/python
Kimish Patel dc2c4d02f9 Add a wrapper to wrap all optimization for mobile. (#35227)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/35227

This wraps.
1. Conv BN folding (not mobile specific)
2. insert XNNPACK conv2d/Linear ops
3. Remove prepacking ops.

Test Plan: Imported from OSS

Differential Revision: D20603562

fbshipit-source-id: ff373af7112c070ec6198bac51845282e09ff1f8
2020-03-25 19:21:14 -07:00
..
init.cpp Add a wrapper to wrap all optimization for mobile. (#35227) 2020-03-25 19:21:14 -07:00
init.h
module_python.h [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
pybind_utils.h [JIT] add id function (#34975) 2020-03-23 17:10:13 -07:00
pybind.h
python_arg_flatten.cpp
python_arg_flatten.h
python_custom_class.cpp [reapply][JIT] Namespaces for TorchBind (#35254) 2020-03-24 00:39:48 -07:00
python_custom_class.h
python_interpreter.cpp Delete OperatorOptions, absorb AliasAnalysisKind into FunctionSchema. (#34588) 2020-03-11 20:59:46 -07:00
python_ir.cpp [jit] make Future type annotation available in Python (#27637) 2020-03-24 14:36:05 -07:00
python_ir.h
python_ivalue.h
python_sugared_value.cpp [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
python_sugared_value.h [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
python_tracer.cpp [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
python_tracer.h [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
python_tree_views.cpp [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
python_tree_views.h [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
script_init.cpp [jit] Include call stack in OSError message (#34669) 2020-03-18 15:10:23 -07:00
script_init.h [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
update_graph_executor_opt.cpp
update_graph_executor_opt.h