projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: dc5ce22) | patch
[fx2trt] Factor out add_matrix_multiply_layer
authorJerry Zhang <jerryzh@fb.com>
Mon, 16 Aug 2021 21:07:43 +0000 (14:07 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Mon, 16 Aug 2021 21:13:37 +0000 (14:13 -0700)
commita12b371f7c0b0180b4661d25c1a2981863d693d1
tree41c2ca0f5c110d5aefd06c7f3b1fc4f17974fb41tree | snapshot
parentdc5ce22a1a8bd075141f7310fa6cb503db4d6e08commit | diff
[fx2trt] Factor out add_matrix_multiply_layer

Summary: Factor out the function so that it can be reused in future diffs

Test Plan: buck run mode/opt caffe2/torch/fb/fx2trt:test_matmul

Reviewed By: 842974287

Differential Revision: D30322823

fbshipit-source-id: 069b945de2c744cdbcca1618b62827692dfb4174
torch/fx/experimental/fx2trt/converters/acc_ops_converters.py diff | blob | history
Domain: Machine Learning / ML Framework;