pytorch/caffe2/utils
Xiaomeng Yang 0fd1dc45c0 Optimize LayerNormOp (#17604)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17604

Optimize LayerNormOp

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

Reviewed By: houseroad

Differential Revision: D14274175

fbshipit-source-id: a7aa263a1b0eb109682d2be99306e7b2cdcc0faf
2019-03-08 17:38:14 -08:00
..
hip Change hip filename extension to .hip (#14036) 2018-11-16 11:55:59 -08:00
math Optimize LayerNormOp (#17604) 2019-03-08 17:38:14 -08:00
threadpool Open registration for c10 thread pool (#17788) 2019-03-08 15:38:41 -08:00
bench_utils.cc
bench_utils.h
cast.h
cast_test.cc
cblas.h
CMakeLists.txt Open registration for c10 thread pool (#17788) 2019-03-08 15:38:41 -08:00
conversions.h Document hip-clang and its __HIP__ macro (#16771) 2019-02-05 15:13:52 -08:00
cpu_neon.h
cpuid.cc
cpuid.h
cpuid_test.cc
eigen_utils.h Export PyTorch erf to ONNX Erf and add Caffe2 Erf operator 2019-01-17 09:18:08 -08:00
fatal_signal_asan_no_sig_test.cc Windows shared build (#13550) 2018-11-16 12:16:28 -08:00
filler.h Delete Tensor::swap(), replace with pointer swap (#12730) 2019-01-25 08:25:07 -08:00
fixed_divisor.h Document hip-clang and its __HIP__ macro (#16771) 2019-02-05 15:13:52 -08:00
fixed_divisor_test.cc
GpuBitonicSort.cuh
GpuDefs.cuh Added HIP top_k operator (#13747) 2018-11-08 20:14:53 -08:00
GpuScanUtils.cuh Separate affine_channel from math and optimize it (#16135) 2019-01-18 22:40:16 -08:00
map_utils.h
math-detail.h
math.h Separate reduce functions from math (#16929) 2019-02-13 17:50:47 -08:00
math_cpu.cc Separate reduce functions from math (#16929) 2019-02-13 17:50:47 -08:00
math_gpu.cu Separate gpu reduce functions (#17146) 2019-02-20 14:49:01 -08:00
math_gpu_test.cc Separate reduce functions from math (#16929) 2019-02-13 17:50:47 -08:00
math_test.cc Separate reduce functions from math (#16929) 2019-02-13 17:50:47 -08:00
murmur_hash3.cc
murmur_hash3.h
proto_convert.cc
proto_convert.h
proto_utils.cc Move files to/from c10/core and c10/util (#15316) 2019-01-10 16:22:22 -08:00
proto_utils.h Add back HIP support to async net (#13400) 2018-10-31 17:52:36 -07:00
proto_utils_test.cc
proto_wrap.cc
proto_wrap.h
signal_handler.cc use C10_MOBILE/ANDROID/IOS (#15363) 2019-01-09 15:08:20 -08:00
signal_handler.h
simple_queue.h Fix issues under caffe round 1 2019-01-23 19:04:59 -08:00
simple_queue_test.cc
smart_tensor_printer.cc Renaming meta() to dtype() - 2/2 (#13334) 2018-10-30 18:24:30 -07:00
smart_tensor_printer.h
smart_tensor_printer_test.cc Kill more weird constructors on Tensor 2018-11-04 16:54:49 -08:00
string_utils.cc
string_utils.h Catch exceptions in bound_shape_inference (#17775) 2019-03-08 13:18:28 -08:00
zmq_helper.h