• Louis Feng's avatar
    [Dynamic Shape] Moving BatchDot to Core Op (#2691) · cc8dd452
    Louis Feng authored
    * batch dot WIP.
    
    * cpu backend refactor and unit tests pass.
    
    * WIP.
    
    * batch dot interpreter impelementation.
    
    * minor clean up.
    
    * more clean up.
    
    * patching the gpu backends.
    
    * added more tests, fixes, etc.
    
    * fixed compile error.
    
    * renamed batch dot to batch matmul.
    
    * refactor WIP.
    
    * fixes some tests and formating.
    
    * more fixes.
    cc8dd452
autodiff.in.cpp 70.9 KB