pytorch/caffe2/utils
Xiaomeng Yang 2ce39de3fc Add elementwise_affine for layer_norm_op (#19713)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19713

Add elementwise_affine for layer_norm_op

Reviewed By: houseroad

Differential Revision: D15075454

fbshipit-source-id: e8a7d3da1c81e49fa55323f5e74a68bc4ef8d83f
2019-04-26 17:20:01 -07:00
..
hip Change hip filename extension to .hip (#14036) 2018-11-16 11:55:59 -08:00
math Add elementwise_affine for layer_norm_op (#19713) 2019-04-26 17:20:01 -07:00
threadpool caffe2 | Use _aligned_free in WorkerPool destruction (#19751) 2019-04-25 14:54:50 -07:00
bench_utils.cc wipe cache with writes (#12279) 2018-10-03 17:12:23 -07:00
bench_utils.h build changes to make cpu unified build working. (#10504) 2018-08-15 17:22:36 -07:00
cast_test.cc Update from facebook (#7855) 2018-05-29 11:38:02 -07:00
cast.h Update from facebook (#7855) 2018-05-29 11:38:02 -07:00
cblas.h
CMakeLists.txt Update math::Transpose to support tensor with size > 2G (#17670) 2019-03-20 18:22:21 -07:00
conversions.h caffe2 | Windows compat fixes 2019-04-23 14:30:19 -07:00
cpu_neon.h [caffe2] Use both __ARM_NEON__ and __ARM_NEON macros (#6697) 2018-04-18 17:45:47 -04:00
cpuid_test.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
cpuid.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
cpuid.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
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_test.cc Use FixedDivisor in Reduce and Broadcast CUDA kernels (#9072) 2018-07-01 00:25:34 -07:00
fixed_divisor.h Document hip-clang and its __HIP__ macro (#16771) 2019-02-05 15:13:52 -08:00
GpuBitonicSort.cuh Manually applying cudnn5 pull request. 2018-01-02 15:31:33 -08:00
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 Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
math_cpu.cc Move math::Axpy function to elementwise lib (#18316) 2019-03-26 12:19:19 -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_gpu.cu optimize BatchMatmulOp (#18612) 2019-04-23 15:34:59 -07:00
math_test.cc Update math::Transpose to support tensor with size > 2G (#17670) 2019-03-20 18:22:21 -07:00
math-detail.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
math.h Move math::Axpy function to elementwise lib (#18316) 2019-03-26 12:19:19 -07:00
murmur_hash3.cc Remove core and util warnings (#8239) 2018-06-07 09:10:33 -07:00
murmur_hash3.h
proto_convert.cc New serialization format (#12384) 2018-10-16 16:36:58 -07:00
proto_convert.h New serialization format (#12384) 2018-10-16 16:36:58 -07: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_utils.cc Add support for getting TensorProto argument (#18364) 2019-04-02 20:58:28 -07:00
proto_utils.h Fix the return value of ParseFromString (#19262) 2019-04-15 12:39:29 -07:00
proto_wrap.cc New Serialization Proto 2018-09-11 10:55:43 -07:00
proto_wrap.h build changes to make cpu unified build working. (#10504) 2018-08-15 17:22:36 -07:00
signal_handler.cc use C10_MOBILE/ANDROID/IOS (#15363) 2019-01-09 15:08:20 -08:00
signal_handler.h More changes for hidden visibility (#10692) 2018-08-21 13:39:57 -07:00
simple_queue_test.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
simple_queue.h Fix issues under caffe round 1 2019-01-23 19:04:59 -08:00
smart_tensor_printer_test.cc Kill more weird constructors on Tensor 2018-11-04 16:54:49 -08:00
smart_tensor_printer.cc Renaming meta() to dtype() - 2/2 (#13334) 2018-10-30 18:24:30 -07:00
smart_tensor_printer.h More changes for hidden visibility (#10692) 2018-08-21 13:39:57 -07:00
string_utils.cc Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
string_utils.h Add backward pass to infer single missing input shape for Concat opportunitiscally (#18911) 2019-04-05 10:11:58 -07:00
zmq_helper.h Unify all *_EXPORT and *_IMPORT macros across c++ backend (#12019) 2018-09-25 17:41:05 -07:00