projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: a97015f) | patch
Add plugin for linalg norm operation (#64611)
authorShirong Wu <shirong@fb.com>
Wed, 8 Sep 2021 21:29:33 +0000 (14:29 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Wed, 8 Sep 2021 21:33:20 +0000 (14:33 -0700)
commitcc0565326ce098236507b13c2bf2d5b48f64fba3
treecf8cd8a9a1dae0ad041a485975db5b43ba69ab29tree | snapshot
parenta97015f22cd5988b16c0eacaa654dec5d7a5a93bcommit | diff
Add plugin for linalg norm operation (#64611)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/64611

Add plugin for torch.linalg.norm, this plugin correctly only support norm operation without batch_size change, so vector input or matrix input with dim including '0' is not supported with this plugin.

Test Plan: Unit test

Reviewed By: 842974287

Differential Revision: D30525958

fbshipit-source-id: 0d66b60a390bb6235166e5a80390090d0acf691a
torch/fx/experimental/fx_acc/acc_ops.py diff | blob | history
Domain: Machine Learning / ML Framework;