https://github.com/MycChiu/fast-LayerNorm-TF
Tip revision: 3521bc41d622a3a772089cba274aedd144163860 authored by mycchiu on 10 March 2017, 02:02:27 UTC
updated layer_norm_bench_mark.py
updated layer_norm_bench_mark.py
Tip revision: 3521bc4
File | Mode | Size |
---|---|---|
images | ||
.gitignore | -rw-r--r-- | 1.0 KB |
LICENSE | -rw-r--r-- | 1.0 KB |
layer_norm_bench_mark.py | -rw-r--r-- | 4.4 KB |
layer_norm_fused_grad_op.cc | -rw-r--r-- | 11.6 KB |
layer_norm_fused_layer.py | -rw-r--r-- | 6.2 KB |
layer_norm_fused_op.cc | -rw-r--r-- | 10.8 KB |
layer_norm_fused_op.h | -rw-r--r-- | 989 bytes |
layer_norm_fused_op.so | -rwxr-xr-x | 2.8 MB |
layer_norm_fused_op_gpu.cu.cc | -rw-r--r-- | 46.3 KB |
layer_norm_fused_test.py | -rw-r--r-- | 8.3 KB |
makefile | -rw-r--r-- | 863 bytes |
makefile_maxwell | -rw-r--r-- | 791 bytes |
readme.md | -rw-r--r-- | 5.6 KB |
register_ops.cc | -rw-r--r-- | 4.4 KB |