* Fix infer type of kernel in dense. * - Moving the check of weight being nullptr up as it is needed in both the branches now. - Adding test case for validating that data dtype and kernel dtypes can be different. * - Fix the dtype check for weight. If the weight is not present then we will use the data dtype.
Name |
Last commit
|
Last update |
---|---|---|
.. | ||
api | Loading commit data... | |
arithmetic | Loading commit data... | |
autotvm | Loading commit data... | |
codegen | Loading commit data... | |
common | Loading commit data... | |
contrib | Loading commit data... | |
lang | Loading commit data... | |
node | Loading commit data... | |
op | Loading commit data... | |
pass | Loading commit data... | |
relay | Loading commit data... | |
runtime | Loading commit data... | |
schedule | Loading commit data... | |
README.md | Loading commit data... |