• Rob Earhart's avatar
    ReshapeElimination -> PlaidML; fixup ImplicitBcast · 1144bf39
    Rob Earhart authored
    The PrefixReshapeElimination pass doesn't work correctly if the network's operation's input and output shape
    compatibility is being validated; an implicit broadcast operation is required to get the shapes right.  This
    change moves the implementation back to PlaidML (since it's seeming less generally useful, and PlaidML has an
    implicit broadcast operation), and fixes both it and the implicit broadcast pass to only apply when the
    downstream operation is an elementwise operation (since PlaidML doesn't support automatic broadcasting for
    contractions).
    1144bf39
Name
Last commit
Last update
.ci Loading commit data...
cmake Loading commit data...
contrib/docker Loading commit data...
doc Loading commit data...
licenses Loading commit data...
maint Loading commit data...
python Loading commit data...
src Loading commit data...
test Loading commit data...
.clang-format Loading commit data...
.gitattributes Loading commit data...
.gitignore Loading commit data...
.travis.yml Loading commit data...
ABOUT.md Loading commit data...
CMakeLists.txt Loading commit data...
CODEOWNERS Loading commit data...
CONTRIB.md Loading commit data...
LICENSE Loading commit data...
README.md Loading commit data...
VERSION.in Loading commit data...
changes.md Loading commit data...
ecosystem-overview.md Loading commit data...