Junjie Bai
c9f380df02
Add aten mkldnn linear operator
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/19210
Reviewed By: dzhulgakov
Differential Revision: D14901641
fbshipit-source-id: 8fa68b9941fd93cea0f313a828cba34c5c81ae11
2019-04-26 13:41:57 -07:00
Junjie Bai
48b81da4cb
Add aten mkldnn view operator
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/19209
Reviewed By: dzhulgakov
Differential Revision: D14894545
fbshipit-source-id: 69455184811de1d1444b5d494e4a9d8c83301431
2019-04-26 13:41:54 -07:00
Junjie Bai
61d5a8dded
Add aten mkldnn add operator
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/19207
Reviewed By: dzhulgakov
Differential Revision: D14889477
fbshipit-source-id: 2c5e5ea5dfc26a9c9a172c5fa2c6d7584b167e16
2019-04-26 13:41:51 -07:00
Junjie Bai
fb53c189b3
Add aten mkldnn batch_norm operator
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/19206
Reviewed By: dzhulgakov
Differential Revision: D14887205
fbshipit-source-id: ea00c9e3205c449d08ab29535309164f951aab95
2019-04-26 13:41:48 -07:00
Junjie Bai
4864000e55
Add aten mkldnn ops: relu, max_pool2d and avg_pool2d
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/19205
Reviewed By: dzhulgakov
Differential Revision: D14850598
fbshipit-source-id: 5bbd5909c06df9c980de680ffb81bf772766c0ba
2019-04-26 13:41:44 -07:00
Junjie Bai
3445020ca3
Add aten mkldnn conv2d operator
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/19204
Reviewed By: dzhulgakov
Differential Revision: D14857513
fbshipit-source-id: 1172c9785e5a17a7d7360474551bdc7a511b3f2f
2019-04-26 13:41:41 -07:00
jgong5
3ad710b837
Add MKL-DNN Tensor ( #17748 )
...
Summary:
This is a minimalist PR to add MKL-DNN tensor per discussion from Github issue: https://github.com/pytorch/pytorch/issues/16038
Ops with MKL-DNN tensor will be supported in following-up PRs to speed up imperative path.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17748
Reviewed By: dzhulgakov
Differential Revision: D14614640
Pulled By: bddppq
fbshipit-source-id: c58de98e244b0c63ae11e10d752a8e8ed920c533
2019-04-08 21:41:38 -07:00
