"vscode:/vscode.git/clone" did not exist on "9d7b5d6c91b0686619010f5355d7eb44c856a95d"
  • ngimel's avatar
    Clean up layer norm tests (#418) · 3ef01fae
    ngimel authored
    * Bug fix for non-affine layer-norm + add backward unit test
    
    * clean up tests and add tests for a large batch
    3ef01fae
layer_norm_cuda_kernel.cu 24.4 KB