projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: c9d7ead) | patch
Add backend checks for batch norm (#15955)
authorvishwakftw <cs15btech11043@iith.ac.in>
Fri, 11 Jan 2019 19:19:56 +0000 (11:19 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Fri, 11 Jan 2019 19:28:45 +0000 (11:28 -0800)
commite07cca13123446d61576a2f54b9c5a680ff9f364
tree3f316f0a9ce5501743b7a25c27b043e417fcc9c7tree | snapshot
parentc9d7ead0c462f4a0b21e8717fc564e43b2827738commit | diff
Add backend checks for batch norm (#15955)

Summary:
Fixes #15826

Changelog:
- Add backend checks in `batch_norm_cpu` and `batch_norm_cuda`
- Modify check in `checkBackend` to pass on undefined tensors.

Differential Revision: D13636410

Pulled By: soumith

fbshipit-source-id: 3b1cfe5ca8b7c0346569077163503065e75c2659
aten/src/ATen/TensorUtils.cpp diff | blob | history
aten/src/ATen/native/Normalization.cpp diff | blob | history
aten/src/ATen/native/cuda/Normalization.cuh diff | blob | history
Domain: Machine Learning / ML Framework;