3 Commits

Author	SHA1	Message	Date
Radhakrishnan Venkataramani	3397919dcf	Rowwise Prune op (Add the test to OSS run_test), Make the op private. (#46131) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/46131 Refer to the title. Test Plan: `buck test caffe2/test:pruning` Reviewed By: raghuramank100 Differential Revision: D24230472 fbshipit-source-id: 8f0a83446c23fdf30d0313b8c3f5ff1a463b50c7	2021-01-29 06:08:18 -08:00
Xiang Gao	20ac736200	Remove py2 compatible future imports (#44735) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/44735 Reviewed By: mruberry Differential Revision: D23731306 Pulled By: ezyang fbshipit-source-id: 0ba009a99e475ddbe22981be8ac636f8a1c8b02f	2020-09-16 12:55:57 -07:00
Radhakrishnan Venkataramani	8032dbc117	Add Rowwise Prune PyTorch op (#42708) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/42708 Add rowwise prune pytorch op. This operator introduces sparsity to the 'weights' matrix with the help of the importance indicator 'mask'. A row is considered important and not pruned if the mask value for that particular row is 1(True) and not important otherwise. Test Plan: buck test caffe2/torch/fb/sparsenn:test -- rowwise_prune buck test caffe2/test:pruning Reviewed By: supriyar Differential Revision: D22849432 fbshipit-source-id: 456f4f77c04158cdc3830b2e69de541c7272a46d	2020-08-27 15:16:23 -07:00