projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 60241e9) | patch
Optimize SpatialBN on GPU (#16202)
authorXiaomeng Yang <yangxm@fb.com>
Thu, 24 Jan 2019 10:50:35 +0000 (02:50 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 24 Jan 2019 10:55:10 +0000 (02:55 -0800)
commit45c3cc9174f707883664e35a2d95d08af3ad3251
tree2734c4819018a7c1154f642e426c72c108707cfatree | snapshot
parent60241e94b3e71f7637230648dcbbd068b71030b4commit | diff
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
caffe2/operators/hip/spatial_batch_norm_op_miopen.hip diff | blob | history
caffe2/operators/spatial_batch_norm_gradient_op.cc diff | blob | history
caffe2/operators/spatial_batch_norm_op.cc diff | blob | history
caffe2/operators/spatial_batch_norm_op.cu diff | blob | history
caffe2/operators/spatial_batch_norm_op.h diff | blob | history
caffe2/operators/spatial_batch_norm_op_cudnn.cc [moved from caffe2/operators/spatial_batch_norm_op_cudnn.cu with 99% similarity] diff | blob | history
caffe2/operators/spatial_batch_norm_op_gpu_impl.cuh [deleted file] blob | history
Domain: Machine Learning / ML Framework;