diff options
author | Richard Zou <zou3519@users.noreply.github.com> | 2018-01-23 21:32:40 -0500 |
---|---|---|
committer | Edward Z. Yang <ezyang@mit.edu> | 2018-01-23 21:32:40 -0500 |
commit | bc11511cda5d8985866dd56c2a67e6c34e5eb074 (patch) | |
tree | 777c367202ff82062fc54a3857d3f10e326ef279 /tools/autograd | |
parent | 23dc8acbc8a6c51e0a75e9933fba27fa66e1dbb1 (diff) | |
download | pytorch-bc11511cda5d8985866dd56c2a67e6c34e5eb074.tar.gz pytorch-bc11511cda5d8985866dd56c2a67e6c34e5eb074.tar.bz2 pytorch-bc11511cda5d8985866dd56c2a67e6c34e5eb074.zip |
Restore sparse variable transpose_() and t_() (#4779)
* Restore sparse variable transpose_() and t_()
* Add dimension wrapping to transpose_, t_
* Don't expose sparse_raw_resize_ to python
Diffstat (limited to 'tools/autograd')
-rw-r--r-- | tools/autograd/gen_python_functions.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tools/autograd/gen_python_functions.py b/tools/autograd/gen_python_functions.py index 994576c67c..47bdb2baa4 100644 --- a/tools/autograd/gen_python_functions.py +++ b/tools/autograd/gen_python_functions.py @@ -16,6 +16,7 @@ CodeTemplate = import_module('code_template', 'aten/src/ATen/code_template.py'). SKIP_PYTHON_BINDINGS = [ 'alias', 'contiguous', 'clamp.*', 'is_cuda', 'is_sparse', 'size', 'stride', '.*_backward', '.*_backward_out', '.*_forward', '.*_forward_out', + 'sparse_raw_resize_', ] PY_VARIABLE_METHODS_CPP = CodeTemplate.from_file(template_path + '/python_variable_methods.cpp') |