• Pruthvi's avatar
    Implementation of CrossEntropy and CrossEntropyBackprop as fused Op's (#3818) · a1a8a7e3
    Pruthvi authored
    * - Implementaion of CrossEntropy and CrossEntropyBackprop as fused Op's
    
    * - unit test case for CE fprop
    - fix bug in decompose_op
    
    * WIP debug PDPD unit test failure
    
    * fixed broadcasting issue
    
    * -fix bdcast issue for multi dim tensor
    
    * utilities to restore the original tensor shape
    
    * i) style-fix ii) rename variables
    
    * - unit test for multiple dimensions ii) refactor create_mask to seperate function
    
    * - fixed unit tests
    
    * fix style
    
    * set output element type to dynamic in pre_validate and infer shape
    
    * disable ce with one hot unit test on PlaidML
    
    * add CE op to fused_op_tbl
    
    * - add serialzier support for CE and CE Backprop
    a1a8a7e3
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...