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
..
broadcast.cc	Separate elementwise level2 math functions (#16753)	2019-02-07 18:38:26 -08:00
broadcast.cu	Separate elementwise level2 math functions (#16753)	2019-02-07 18:38:26 -08:00
broadcast.h	Separate elementwise level2 math functions (#16753)	2019-02-07 18:38:26 -08:00
elementwise.cc	Optimize Scale function (#44913)	2020-09-18 14:31:33 -07:00
elementwise.cu	Optimize Scale function (#44913)	2020-09-18 14:31:33 -07:00
elementwise.h	Optimize Scale function (#44913)	2020-09-18 14:31:33 -07:00
half_utils.h	Separate elementwise level2 math functions (#16753)	2019-02-07 18:38:26 -08:00
reduce.cc	Add elementwise_affine for layer_norm_op (#19713)	2019-04-26 17:20:01 -07:00
reduce.cu	Add elementwise_affine for layer_norm_op (#19713)	2019-04-26 17:20:01 -07:00
reduce.cuh	Separate gpu reduce functions (#17146)	2019-02-20 14:49:01 -08:00
reduce.h	Separate reduce functions from math (#16929)	2019-02-13 17:50:47 -08:00
transpose.cc	[caffe2] fix Transpose2D calls in NHWC<->NCHW (#34625)	2020-03-16 15:18:44 -07:00
transpose.cu	Fix bug in caffe2 transpose on GPU (#22233)	2019-06-26 11:33:25 -07:00
transpose.h	Update math::Transpose to support tensor with size > 2G (#17670)	2019-03-20 18:22:21 -07:00
utils.cc	optimize BatchMatmulOp (#18612)	2019-04-23 15:34:59 -07:00
utils.h	[caffe2] use Clang identification macro in various places (#33574)	2020-02-20 15:16:11 -08:00