pytorch/torch/csrc/jit/python
goldenxuett 21a4be34cd [JIT] Enchance training ops check to be more inclusive and account for possible pybind exceptions (#81782)
- Modified is_mutable python binding to accept a string instead of a string_view for better python compatibility.
- Modified argument value adding python bindings to deal with input/self edge case due to inconsistencies in how the first variable is named.
- Modified _is_alias_of and created _contains_alias_of python bindings to accurately find out if values are aliasing, or contain an alias.
- Fixed is_mutable implementation to cover all ops that have mutable optional arguments. (These are all the ops that have the optional arguments 'running_mean' and 'running_var' along with either 'train', 'training' or 'use_input_stats.'
Pull Request resolved: https://github.com/pytorch/pytorch/pull/81782
Approved by: https://github.com/davidberard98
2022-07-20 22:09:54 +00:00
..
init.cpp [JIT] Enchance training ops check to be more inclusive and account for possible pybind exceptions (#81782) 2022-07-20 22:09:54 +00:00
init.h
module_python.h
pybind_utils.cpp Add support for Generator conversion to/from IValue (#81697) 2022-07-19 16:50:10 +00:00
pybind_utils.h Add support for Generator conversion to/from IValue (#81697) 2022-07-19 16:50:10 +00:00
pybind.h
python_arg_flatten.cpp [ONNX] Support optional type (#68793) (#73284) 2022-05-04 20:24:30 +00:00
python_arg_flatten.h
python_custom_class.cpp
python_custom_class.h
python_dict.cpp
python_dict.h
python_interpreter.cpp
python_ir.cpp canonicalize includes of form <aten/src/ATen/...> 2022-06-16 17:46:45 +00:00
python_ir.h
python_ivalue.h
python_list.cpp
python_list.h Fix sign-compare violations in python_list.h 2022-04-01 19:15:51 +00:00
python_sugared_value.cpp turn on -Werror=unused-variable in our Bazel CPU build 2022-06-11 02:46:34 +00:00
python_sugared_value.h [JIT] Support for ParameterDict getattr 2022-05-13 01:03:25 +00:00
python_tracer.cpp
python_tracer.h
python_tree_views.cpp Reland "Make debug_pkl smaller by only emitting unique traces." (#73368) 2022-04-18 22:34:21 +00:00
python_tree_views.h
script_init.cpp Expose _export_data from C++ to Python (#79207) 2022-06-10 00:41:33 +00:00
script_init.h
update_graph_executor_opt.cpp
update_graph_executor_opt.h