pytorch/caffe2/core
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 Replace caffe2::DeviceGuard with c10::cuda::CUDAGuard (#17623) 2019-03-06 10:48:15 -08:00
nomnigraph Fix out-of-topological-order issue in Nomnigraph (#19458) 2019-04-19 12:19:39 -07:00
__init__.py re-enable copy of python files, but be careful that the copy is only … (#14982) 2018-12-11 16:54:08 -08:00
allocator.cc unify c2 and TH allocator (#16892) 2019-02-12 21:16:34 -08:00
allocator.h unify c2 and TH allocator (#16892) 2019-02-12 21:16:34 -08:00
asan.h
blob_gpu_test.cc Replace caffe2::DeviceGuard with c10::cuda::CUDAGuard (#17623) 2019-03-06 10:48:15 -08:00
blob_serialization_gpu.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
blob_serialization.cc Replace caffe2::DeviceGuard with c10::cuda::CUDAGuard (#17623) 2019-03-06 10:48:15 -08:00
blob_serialization.h fix -Wsign-compare warnings for some files inside c2 (#18123) 2019-03-19 10:39:20 -07:00
blob_serializer_base.h Serialization takes pointers instead of Blob (#11925) 2018-10-17 11:50:34 -07:00
blob_stats.cc codemod -d caffe2 --extensions cc,h CaffeTypeId TypeIdentifier (#10166) 2018-08-02 11:54:30 -07:00
blob_stats.h Fix include paths for typeid.h (#13689) 2018-11-14 18:04:09 -08:00
blob_test.cc Apply modernize-use-override (2nd iteration) 2019-02-14 16:52:57 -08:00
blob.h Adding quantized tensor shape/type info support for caffe2=>glow in caffe2 side (#18621) 2019-03-31 17:42:27 -07:00
c10_operator.h Use string based schema for exposing caffe2 ops (#19287) 2019-04-18 02:04:50 -07:00
CMakeLists.txt Change hip filename extension to .hip (#14036) 2018-11-16 11:55:59 -08:00
common_cudnn.cc
common_cudnn.h Unify all *_EXPORT and *_IMPORT macros across c++ backend (#12019) 2018-09-25 17:41:05 -07:00
common_gpu.cc Delete caffe2_cuda_full_device_control (#14283) 2018-11-29 18:33:22 -08:00
common_gpu.h Replace caffe2::DeviceGuard with c10::cuda::CUDAGuard (#17623) 2019-03-06 10:48:15 -08:00
common_omp.h
common_test.cc Windows shared build (#13550) 2018-11-16 12:16:28 -08:00
common.cc
common.h use C10_MOBILE/ANDROID/IOS (#15363) 2019-01-09 15:08:20 -08:00
context_base.cc Back out "[pytorch][PR] Revert "Move CreateContext to global registry (#11688)"" (#12121) 2018-10-01 11:10:46 -07:00
context_base.h Revert "Move CreateContext to global registry (#11688)" (#12049) 2018-09-25 10:13:43 -07:00
context_gpu_test.cc Replace caffe2::DeviceGuard with c10::cuda::CUDAGuard (#17623) 2019-03-06 10:48:15 -08:00
context_gpu.cu Caffe2 shouldn't fail if CUDA peer access is already enabled 2019-04-24 13:22:27 -07:00
context_gpu.h Replace caffe2::DeviceGuard with c10::cuda::CUDAGuard (#17623) 2019-03-06 10:48:15 -08:00
context_test.cc unify c2 and TH allocator (#16892) 2019-02-12 21:16:34 -08:00
context.cc Change Tensor::CopyFrom to a simple double dispatch (#14268) 2018-11-28 15:45:37 -08:00
context.h Fix a lot of C++ build warnings (#16411) 2019-01-31 14:35:56 -08:00
cudnn_wrappers.h Replace caffe2::DeviceGuard with c10::cuda::CUDAGuard (#17623) 2019-03-06 10:48:15 -08:00
db.cc Apply modernize-use-override (2nd iteration) 2019-02-14 16:52:57 -08:00
db.h Change message on unknown db type to be friendly (#17795) 2019-03-08 10:46:24 -08:00
event_cpu.h Delete unnecessary include from allocator.cc/event_cpu.h 2018-09-19 16:45:54 -07:00
event_gpu_test.cc caffe2::DeviceType -> at::DeviceType (#11254) 2018-09-05 16:28:09 -07:00
event_gpu.cc Replace caffe2::DeviceGuard with c10::cuda::CUDAGuard (#17623) 2019-03-06 10:48:15 -08:00
event_test.cc caffe2::DeviceType -> at::DeviceType (#11254) 2018-09-05 16:28:09 -07:00
event.cc Update from facebook (#8384) 2018-06-13 13:10:45 -07:00
event.h Fix issues under caffe round 1 2019-01-23 19:04:59 -08:00
flags.h move flags to c10 (#12144) 2018-10-04 02:09:56 -07:00
graph_test.cc
graph.cc Replace direct include of caffe2.pb.h with an intermediary header caffe2_pb.h (#10946) 2018-08-28 11:57:08 -07:00
graph.h Replace direct include of caffe2.pb.h with an intermediary header caffe2_pb.h (#10946) 2018-08-28 11:57:08 -07:00
init_denormals.cc add denormal options (ftz and daz) 2018-12-20 17:04:39 -08:00
init_intrinsics_check.cc Using c10 namespace across caffe2. (#12714) 2018-10-17 12:57:19 -07:00
init_omp.cc Using c10 namespace across caffe2. (#12714) 2018-10-17 12:57:19 -07:00
init_test.cc Allow for registration after GlobalInit (#15876) 2019-01-10 09:35:33 -08:00
init.cc Using c10 namespace across caffe2. (#12714) 2018-10-17 12:57:19 -07:00
init.h Allow for registration after GlobalInit (#15876) 2019-01-10 09:35:33 -08:00
int8_serialization.cc Fix include paths for typeid.h (#13689) 2018-11-14 18:04:09 -08:00
logging.h Moving logging from caffe2 to c10. (#12881) 2018-10-19 20:22:08 -07:00
macros.h
macros.h.in use C10_MOBILE/ANDROID/IOS (#15363) 2019-01-09 15:08:20 -08:00
memonger.cc Windows shared build (#13550) 2018-11-16 12:16:28 -08:00
memonger.h Replace direct include of caffe2.pb.h with an intermediary header caffe2_pb.h (#10946) 2018-08-28 11:57:08 -07:00
module_test.cc
module.cc
module.h Fix include paths for typeid.h (#13689) 2018-11-14 18:04:09 -08:00
net_async_base.cc Open registration for c10 thread pool (#17788) 2019-03-08 15:38:41 -08:00
net_async_base.h Open registration for c10 thread pool (#17788) 2019-03-08 15:38:41 -08:00
net_async_scheduling.cc Rethrow exceptions from RunAsync (#15034) 2019-02-20 16:32:24 -08:00
net_async_scheduling.h Reinforce scheduling invariants (#17132) 2019-02-20 16:32:23 -08:00
net_async_task_future.cc Back out "Revert D13043261: [caffe2] Task graph and task future abstractions in executor" 2018-12-10 19:30:58 -08:00
net_async_task_future.h Back out "Revert D13043261: [caffe2] Task graph and task future abstractions in executor" 2018-12-10 19:30:58 -08:00
net_async_task_graph.cc Back out "Revert D13043261: [caffe2] Task graph and task future abstractions in executor" 2018-12-10 19:30:58 -08:00
net_async_task_graph.h Back out "Revert D13043261: [caffe2] Task graph and task future abstractions in executor" 2018-12-10 19:30:58 -08:00
net_async_task.cc Back out "Revert D13043261: [caffe2] Task graph and task future abstractions in executor" 2018-12-10 19:30:58 -08:00
net_async_task.h Back out "Revert D13043261: [caffe2] Task graph and task future abstractions in executor" 2018-12-10 19:30:58 -08:00
net_async_tracing_test.cc AsyncNet: option for time based tracing and trace path (#13440) 2018-11-08 11:34:34 -08:00
net_async_tracing.cc Windows shared build (#13550) 2018-11-16 12:16:28 -08:00
net_async_tracing.h AsyncNet: option for time based tracing and trace path (#13440) 2018-11-08 11:34:34 -08:00
net_dag_utils_test.cc New chaining/partitioning algorithm for async_scheduling for inference (#11957) 2018-10-08 12:24:52 -07:00
net_dag_utils.cc Back out "[nomnigraph][executor] computeChains with nomnigraph" (#15451) 2018-12-21 11:09:27 -08:00
net_dag_utils.h Back out "[nomnigraph][executor] computeChains with nomnigraph" (#15451) 2018-12-21 11:09:27 -08:00
net_gpu_test.cc Remove unused executors, part 3 (#14199) 2018-11-26 19:10:43 -08:00
net_parallel.cc Open registration for c10 thread pool (#17788) 2019-03-08 15:38:41 -08:00
net_parallel.h TBB task graph (#15041) 2018-12-10 21:35:04 -08:00
net_simple_refcount_test.cc Remove unused executors, part 3 (#14199) 2018-11-26 19:10:43 -08:00
net_simple_refcount.cc net_simple_refcount type to help experimentation with dynamic allocation. (#13370) 2018-10-31 15:59:16 -07:00
net_simple_refcount.h net_simple_refcount type to help experimentation with dynamic allocation. (#13370) 2018-10-31 15:59:16 -07:00
net_simple.cc Using c10 namespace across caffe2. (#12714) 2018-10-17 12:57:19 -07:00
net_simple.h Move registry fully to c10 (#12077) 2018-09-27 03:09:54 -07:00
net_test.cc Better handling of net errors in prof_dag counters (#17384) 2019-02-22 18:38:31 -08:00
net.cc Allow empty net type (#19154) 2019-04-11 12:43:07 -07:00
net.h Open registration for c10 thread pool (#17788) 2019-03-08 15:38:41 -08:00
numa.cc Move numa.{h, cc} to c10/util (#15024) 2018-12-12 12:21:10 -08:00
numa.h Move numa.{h, cc} to c10/util (#15024) 2018-12-12 12:21:10 -08:00
observer_test.cc Apply modernize-use-override (2nd iteration) 2019-02-14 16:52:57 -08:00
observer.h Unify all *_EXPORT and *_IMPORT macros across c++ backend (#12019) 2018-09-25 17:41:05 -07:00
operator_c10wrapper.cc Move registry fully to c10 (#12077) 2018-09-27 03:09:54 -07:00
operator_c10wrapper.h Add elementwise_affine for layer_norm_op (#19713) 2019-04-26 17:20:01 -07:00
operator_gpu_test.cc caffe2::DeviceType -> at::DeviceType (#11254) 2018-09-05 16:28:09 -07:00
operator_gradient.h Define REGISTER_CPU_GRADIENT_OPERATOR (#12588) 2018-10-22 10:01:02 -07:00
operator_schema_test.cc
operator_schema.cc Add filler for SparseLengthsWeightedSum (#13949) 2018-11-16 11:31:05 -08:00
operator_schema.h Verify def before infer fensor (#18129) 2019-03-22 06:36:25 -07:00
operator_test.cc Remove legacy way of exposing caffe2 operators to PyTorch (#17742) 2019-03-08 10:22:41 -08:00
operator.cc Adding quantized tensor shape/type info support for caffe2=>glow in caffe2 side (#18621) 2019-03-31 17:42:27 -07:00
operator.h Adding quantized tensor shape/type info support for caffe2=>glow in caffe2 side (#18621) 2019-03-31 17:42:27 -07:00
parallel_net_test.cc
plan_executor.cc Renaming size() to numel() - 1/17 2018-10-26 15:36:59 -07:00
plan_executor.h Remove unused feature: num PS tuning 2018-07-11 18:54:45 -07:00
prof_dag_counters.cc Better handling of net errors in prof_dag counters (#17384) 2019-02-22 18:38:31 -08:00
prof_dag_counters.h Better handling of net errors in prof_dag counters (#17384) 2019-02-22 18:38:31 -08:00
qtensor_serialization.cc
qtensor_serialization.h Guard all Caffe2 protobuf string serializations with CAFFE_ENFORCE (fixed reverted bug) (#12848) 2018-10-23 16:21:26 -07:00
qtensor.cc
qtensor.h rest of uses for deprecation of dims() in Tensor (#16118) 2019-01-18 11:52:12 -08:00
scope_guard.h
static_tracepoint_elfx86.h
static_tracepoint.h
stats_test.cc Add CAFFE_STATIC_EVENT to Stats (#9501) 2018-07-19 16:25:59 -07:00
stats.cc
stats.h Add full namespace resolution in CAFFE_DURATION (#12065) 2018-09-26 13:29:18 -07:00
storage.h Move files to/from c10/core and c10/util (#15316) 2019-01-10 16:22:22 -08:00
tensor_impl.h Fix include paths for TensorImpl.h 2018-12-11 21:01:40 -08:00
tensor_int8.cc Fix include paths for typeid.h (#13689) 2018-11-14 18:04:09 -08:00
tensor_int8.h Replace direct include of caffe2.pb.h with an intermediary header caffe2_pb.h (#10946) 2018-08-28 11:57:08 -07:00
tensor.cc Change ReinitializeTensor to use C10_LOG_FIRST_N (#18531) 2019-04-02 21:03:37 -07:00
tensor.h Moving at::Tensor into caffe2::Tensor without bumping refcount (#19388) 2019-04-18 14:13:26 -07:00
test_utils.cc nomnigraph - easy - use new test utils in converter_nomnigraph_test (#15751) 2019-01-14 18:38:38 -08:00
test_utils.h Make BlackBoxPredictor handle networks throwing exceptions (#19080) 2019-04-09 16:42:12 -07:00
timer_test.cc
timer.h Unify all *_EXPORT and *_IMPORT macros across c++ backend (#12019) 2018-09-25 17:41:05 -07:00
transform_test.cc
transform.cc Move registry fully to c10 (#12077) 2018-09-27 03:09:54 -07:00
transform.h Move registry fully to c10 (#12077) 2018-09-27 03:09:54 -07:00
types.cc Corresponding data type for BYTE (#15627) 2019-01-17 00:17:56 -08:00
types.h Remove ATen/Half.h and ATen/core/Half.h forwarding headers. 2019-01-18 10:55:21 -08:00
workspace_test.cc Print blob sizes on fatal signal (#10766) 2018-08-23 13:39:55 -07:00
workspace.cc move flags to c10 (#12144) 2018-10-04 02:09:56 -07:00
workspace.h Fix issues under caffe round 1 2019-01-23 19:04:59 -08:00