Revision 9d1e1e32de1eefb4bcea90e26f7df1dddc972c9c authored by Steven Johnson on 20 July 2021, 00:33:52 UTC, committed by GitHub on 20 July 2021, 00:33:52 UTC
* [hannk] Rewrite FC in terms of Conv2D

FullyConnected is very similar to Conv2D, so rather than maintaining multiple similar implementations, let's translate a FullyConnected node into a Conv2D node (with some Reshape nodes as necessary).

Note that we keep the old FC logic for int16 outputs, as Conv2D doesn't support those yet; if this PR is landed, a followup PR will add that ability to Conv2D, and the existing FC support will be removed entirely.
1 parent bd7ebf5
History
File Mode Size
common
tgz
ubuntu
zip
CMakeLists.txt -rw-r--r-- 7.8 KB

back to top