Skip to content

Commit

Permalink
Specify the operand data type constraints of operations (#646)
Browse files Browse the repository at this point in the history
* Specify the operand data type constraints of operations

Introduce constraints for input operands, either directly (e.g.
input's dataType can only be "float32" or "float16") or indirectly
(e.g. weight's dataType must be the same as input's).

Fixes #283

* gruCell: bundle hiddenState in with other type validations

* Identity should accept all types

* Add reduceMean restriction

* Update gemm to check c data type too

---------

Co-authored-by: Dwayne Robinson <[email protected]>
  • Loading branch information
inexorabletash and fdwr authored Apr 26, 2024
1 parent d57e4ef commit b5677b6
Showing 1 changed file with 83 additions and 29 deletions.
Loading

0 comments on commit b5677b6

Please sign in to comment.