pytorch/caffe2/utils
Kevin Matzen 6d8649dc53 [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
2020-03-16 15:18:44 -07:00
..
hip
math [caffe2] fix Transpose2D calls in NHWC<->NCHW (#34625) 2020-03-16 15:18:44 -07:00
threadpool Added nullptr check for pthradpool_get_threads_count (#34087) 2020-03-04 11:10:53 -08:00
bench_utils.cc
bench_utils.h Lightweight at-most-once logging for API usage (#20745) 2019-05-23 23:17:59 -07:00
cast.h
cast_test.cc
cblas.h
CMakeLists.txt Fixing pthreadpool symbol conflict issue. (#33869) 2020-02-28 21:23:18 -08:00
conversions.h [caffe2] use Clang identification macro in various places (#33574) 2020-02-20 15:16:11 -08:00
cpu_neon.h
cpuid.cc
cpuid.h
cpuid_test.cc
eigen_utils.h
fatal_signal_asan_no_sig_test.cc
filler.h
fixed_divisor.h [caffe2] use Clang identification macro in various places (#33574) 2020-02-20 15:16:11 -08:00
fixed_divisor_test.cc Enable ROCm multi-gpu with Gloo 2019-05-07 09:55:47 -07:00
GpuBitonicSort.cuh
GpuDefs.cuh [caffe2] fix invalid % escape in inline assembly strings (#33554) 2020-02-20 14:31:52 -08:00
GpuScanUtils.cuh
map_utils.h Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
math-detail.h
math.h Move math::Axpy function to elementwise lib (#18316) 2019-03-26 12:19:19 -07:00
math_cpu.cc Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
math_gpu.cu switch to rocThrust for thrust/cub APIs (#25620) 2019-09-03 22:16:30 -07:00
math_gpu_test.cc Update math::Transpose to support tensor with size > 2G (#17670) 2019-03-20 18:22:21 -07:00
math_test.cc Update math::Transpose to support tensor with size > 2G (#17670) 2019-03-20 18:22:21 -07:00
murmur_hash3.cc
murmur_hash3.h
proto_convert.cc
proto_convert.h
proto_utils.cc [net_runner] Get shape info from qtensors (#34321) 2020-03-09 18:34:16 -07:00
proto_utils.h Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
proto_utils_test.cc caffe2 - Util to cleanup external inputs and outputs from a NetDef (#18194) 2019-03-22 11:23:03 -07:00
proto_wrap.cc
proto_wrap.h
signal_handler.cc preprocessor cleanup (#33957) 2020-03-02 13:37:19 -08:00
signal_handler.h Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
simple_queue.h
simple_queue_test.cc
smart_tensor_printer.cc preprocessor cleanup (#33957) 2020-03-02 13:37:19 -08:00
smart_tensor_printer.h
smart_tensor_printer_test.cc
string_utils.cc BlackBoxPredictor OSS part 5: glow transforms 2019-07-23 16:39:23 -07:00
string_utils.h move GetDimFromOrderString to caffe2/core/types.h (#25671) 2019-09-05 04:32:04 -07:00
zmq_helper.h Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00