Moving torch.norm to ATen using TensorIterator (#15414)
authorjiej <jiej@nvidia.com>
Thu, 17 Jan 2019 06:12:13 +0000 (22:12 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 17 Jan 2019 06:15:25 +0000 (22:15 -0800)
commit7c56db73d5a9e1432dabc0231acad63575c3089e
treefd26ce91d87d61ab336165285ef04c28a5dd3f8d
parent55511004d17bd3e0e36e88efa6abdc9a5a03dec1
Moving torch.norm to ATen using TensorIterator (#15414)

Summary:
Adding supports for torch.nomr:
i. multi dimensions for dim
ii. dtype that specifies math/output tensor type
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15414

Differential Revision: D13702022

Pulled By: ezyang

fbshipit-source-id: da2676f2b6aff988889b1539d0de8ecd4946823a
18 files changed:
aten/src/ATen/Declarations.cwrap
aten/src/ATen/core/Tensor.h
aten/src/ATen/core/TensorMethods.h
aten/src/ATen/core/Type.h
aten/src/ATen/core/aten_interned_strings.h
aten/src/ATen/native/LinearAlgebra.cpp
aten/src/ATen/native/ReduceOps.cpp
aten/src/ATen/native/ReduceOps.h
aten/src/ATen/native/SharedReduceOps.h
aten/src/ATen/native/cpu/ReduceOpsKernel.cpp
aten/src/ATen/native/cuda/ReduceOpsKernel.cu
aten/src/ATen/native/native_functions.yaml
test/test_cuda.py
tools/autograd/derivatives.yaml
tools/autograd/templates/Functions.cpp
torch/csrc/jit/passes/shape_analysis.cpp
torch/functional.py
torch/tensor.py