OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
7dd5da2026
pytorch/torch/csrc/jit/python
History
Kimish Patel 7dd5da2026 JIT pass to insert XNNPACK ops (#34048) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/34048

Rewrites the graph to insert xnnpack prepack and packed run ops for
conv2d and linear.

Test Plan:
python test/test_xnnpack_integration.py

Imported from OSS

Differential Revision: D20185658

fbshipit-source-id: c4c073c912ad33e822e7beb4ed86c9f895129d55
2020-03-14 12:53:27 -07:00
..
init.cpp JIT pass to insert XNNPACK ops (#34048) 2020-03-14 12:53:27 -07:00
init.h [jit] do the code reorg (#33851) 2020-02-27 13:02:51 -08:00
module_python.h [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
pybind_utils.h invokeOperatorFromPython: support overloaded operator calling (#34671) 2020-03-13 18:46:03 -07:00
pybind.h [jit] do the code reorg (#33851) 2020-02-27 13:02:51 -08:00
python_arg_flatten.cpp [jit] do the code reorg (#33851) 2020-02-27 13:02:51 -08:00
python_arg_flatten.h [jit] do the code reorg (#33851) 2020-02-27 13:02:51 -08:00
python_custom_class.cpp [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
python_custom_class.h [jit] do the code reorg (#33851) 2020-02-27 13:02:51 -08:00
python_interpreter.cpp Delete OperatorOptions, absorb AliasAnalysisKind into FunctionSchema. (#34588) 2020-03-11 20:59:46 -07:00
python_ir.cpp [CUDA_FUSER] Fork CUDA fuser (#33527) 2020-03-04 20:25:08 -08:00
python_ir.h [jit] do the code reorg (#33851) 2020-02-27 13:02:51 -08:00
python_ivalue.h [jit] do the code reorg (#33851) 2020-02-27 13:02:51 -08:00
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] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
script_init.h [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
update_graph_executor_opt.cpp [jit] do the code reorg (#33851) 2020-02-27 13:02:51 -08:00
update_graph_executor_opt.h [jit] do the code reorg (#33851) 2020-02-27 13:02:51 -08:00