DeepSpeed
0a2ae2ef - Fix the MLP output tensor's shape (#2380)

Commit

2 years ago

Fix the MLP output tensor's shape (#2380)

References

#2380 - Fix the MLP output tensor's shape

Author

arashb

arashb

Parents

Files1

csrc/transformer/inference/csrc
- pt_binding.cpp