TensorComprehensions
TensorComprehensions copied to clipboard
Fix the SpatialBatchNorm C++ test to work with latest ATen
PR https://github.com/facebookresearch/TensorComprehensions/pull/183 disables SpatialBatchNorm test. It has various issues:
- dimension mismatching for weight
- output doesn't match in the specific reference check
this is related to changes in ATen API.
The test seems to pass for me on master...