Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Specify the operand data type constraints of operations (#646)
* 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