pytorch/caffe2/utils/math
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
broadcast.cu
broadcast.h
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
reduce.cc
reduce.cu
reduce.cuh
reduce.h
transpose.cc [caffe2] fix Transpose2D calls in NHWC<->NCHW (#34625) 2020-03-16 15:18:44 -07:00
transpose.cu
transpose.h
utils.cc
utils.h