pytorch
6d8649dc - [caffe2] fix Transpose2D calls in NHWC<->NCHW (#34625)

Commit
4 years ago
[caffe2] fix Transpose2D calls in NHWC<->NCHW (#34625) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/34625 These templated function calls are not specifying the template args correctly. The first arg is the index type, not the array data type. That means, right now it's using `T` as the index type as well, which will break if we do a template specialization for uint8_t. If we omit both, it will correctly infer that the index type is `int` and the data type is `T`. Reviewed By: BIT-silence Differential Revision: D20358728 fbshipit-source-id: 8cbd8eeb14bce602c02eb6fce2cc141f0121fa24
Author
Parents
Loading