Separate affine_channel from math and optimize it (#16135)
authorXiaomeng Yang <yangxm@fb.com>
Sat, 19 Jan 2019 06:37:12 +0000 (22:37 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Sat, 19 Jan 2019 06:40:16 +0000 (22:40 -0800)
commitb436f94b53e261177422fe92680d42f19195d3d0
tree37d276fb86c4f7ee5ba474308c6037d1802af786
parente8b872abe225dfc9f622eaa847e82ad7296ea17b
Separate affine_channel from math and optimize it (#16135)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16135

Separate affine_channel from math and optimize it

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

Reviewed By: houseroad

Differential Revision: D13727606

fbshipit-source-id: 8980af4afadaf964a18a9da581106fe30896a7e9
14 files changed:
caffe2/operators/length_split_op.h
caffe2/operators/stylizer_ops.cc
caffe2/operators/top_k.cu
caffe2/operators/top_k_heap_selection.cuh
caffe2/operators/top_k_radix_selection.cuh
caffe2/utils/CMakeLists.txt
caffe2/utils/GpuScanUtils.cuh
caffe2/utils/math.h
caffe2/utils/math/elementwise.cc [new file with mode: 0644]
caffe2/utils/math/elementwise.cu [new file with mode: 0644]
caffe2/utils/math/elementwise.h [new file with mode: 0644]
caffe2/utils/math_cpu.cc
caffe2/utils/math_gpu.cu
caffe2/utils/math_utils.h