OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
android_fbgemm_off
pytorch/caffe2/utils/math
History
Xiaomeng Yang 29b53b0259 Fix bug in caffe2 transpose on GPU (#22233) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/22233

Fix bug in caffe2 transpose on GPU

Reviewed By: hl475

Differential Revision: D15994973

fbshipit-source-id: 542dc8757b51a6322fffa55826c1d4e32927398d
2019-06-26 11:33:25 -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 Add gelu op (#18992) 2019-04-08 21:58:29 -07:00
elementwise.cu Add gelu op (#18992) 2019-04-08 21:58:29 -07:00
elementwise.h Add gelu op (#18992) 2019-04-08 21:58:29 -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 Update math::Transpose to support tensor with size > 2G (#17670) 2019-03-20 18:22:21 -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 optimize BatchMatmulOp (#18612) 2019-04-23 15:34:59 -07:00