• 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
serializer.cpp 157 KB