Optimize SpatialBN on GPU (#16202)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16202
Optimize SpatialBN on GPU
Reviewed By: houseroad
Differential Revision: D13747581
fbshipit-source-id: 48a885a240ef2a325235e8f89ebbe50e7c780c84