mirror of
https://github.com/ml-explore/mlx.git
synced 2025-08-22 04:56:41 +08:00
updated BN implementation to handle input shape as NLC and NWHC^^
This commit is contained in:
parent
9bf68814a4
commit
a1c06b7d46