pytorch/caffe2/utils
Lingyi Liu 2d884f2263 Optimize Scale function (#44913)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/44913

Pull Request resolved: https://github.com/pytorch/pytorch/pull/18322

Optimize Scale function

i-am-not-moving-c2-to-c10

Test Plan: buck test mode/dbg caffe2/caffe2/python/operator_test:weighted_sum_test

Reviewed By: BIT-silence

Differential Revision: D14575780

fbshipit-source-id: db333a7964581dcaff6e432ff1d6b517ba1a075f
2020-09-18 14:31:33 -07:00
..
hip
math Optimize Scale function (#44913) 2020-09-18 14:31:33 -07:00
threadpool Re-apply PyTorch pthreadpool changes 2020-06-23 19:26:21 -07:00
bench_utils.cc
bench_utils.h
cast_test.cc
cast.h
cblas.h
CMakeLists.txt Compile less legacy code when BUILD_CAFFE2 is set to False (take 2) (#44453) 2020-09-11 16:27:47 -07:00
conversions.h [caffe2] use Clang identification macro in various places (#33574) 2020-02-20 15:16:11 -08:00
cpu_neon.h
cpuid_test.cc
cpuid.cc
cpuid.h [caffe2] Use cpuinfo in perfkernels to simplify build dependency (#36371) 2020-04-10 13:26:34 -07:00
eigen_utils.h
fatal_signal_asan_no_sig_test.cc
filler.h
fixed_divisor_test.cc
fixed_divisor.h [caffe2] use Clang identification macro in various places (#33574) 2020-02-20 15:16:11 -08:00
GpuBitonicSort.cuh
GpuDefs.cuh Initial support for building on Ampere GPU, CUDA 11, cuDNN 8 (#39277) 2020-06-02 10:03:42 -07:00
GpuScanUtils.cuh RIP CUDA <9.2: circleci, aten, and caffe2 (#36846) 2020-05-18 13:41:05 -07:00
map_utils.h Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
math_cpu.cc Histogram Binning Calibration 2020-09-06 17:11:16 -07:00
math_gpu_test.cc
math_gpu.cu RIP CUDA <9.2: circleci, aten, and caffe2 (#36846) 2020-05-18 13:41:05 -07:00
math_test.cc
math-detail.h
math.h
murmur_hash3.cc
murmur_hash3.h
proto_convert.cc
proto_convert.h
proto_utils_test.cc
proto_utils.cc [caffe2][redo] Reimplement RemoveOpsByType with SSA (#41606) 2020-07-17 16:06:43 -07:00
proto_utils.h Logging for ATen op subtype 2020-04-19 23:02:50 -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_test.cc
simple_queue.h
smart_tensor_printer_test.cc
smart_tensor_printer.cc preprocessor cleanup (#33957) 2020-03-02 13:37:19 -08:00
smart_tensor_printer.h
string_utils.cc
string_utils.h Fix out-of-boundary access in caffe2::StartsWith (#36672) 2020-04-15 20:40:59 -07:00
zmq_helper.h Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00