Use C10Tensor in the dispatcher (#15195)
authorSebastian Messmer <messmer@fb.com>
Wed, 9 Jan 2019 04:22:41 +0000 (20:22 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 9 Jan 2019 04:31:43 +0000 (20:31 -0800)
commitd562840910b8743b6ea476a47c4df53a8531fd14
tree71b9ddf1d85c8a697987697f325f807623be5acc
parent8ac55a6812884d76d6116aa72aa7beb4a6bda832
Use C10Tensor in the dispatcher (#15195)

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

This removes the use of caffe2::Tensor or at::Tensor in the c10 dispatcher and only uses C10::Tensor.
It also changes output tensors to be passed as `const Tensor&` instead of `Tensor*` because we otherwise can't forward them in operator_c10wrapper.h.

Reviewed By: ezyang

Differential Revision: D13461640

fbshipit-source-id: 7f79925a7d60f01660a24bbfda47391af0c70ed3
59 files changed:
c10/CMakeLists.txt
c10/core/Tensor.h
c10/core/dispatch/OpSchema.h
c10/test/dispatch/OpSchema_test.cpp
caffe2/core/operator_c10wrapper.h
caffe2/operators/experimental/c10/cpu/add_cpu.cc
caffe2/operators/experimental/c10/cpu/averaged_loss_cpu.cc
caffe2/operators/experimental/c10/cpu/batch_gather_cpu.cc
caffe2/operators/experimental/c10/cpu/batch_matmul_cpu.cc
caffe2/operators/experimental/c10/cpu/cast_cpu.cc
caffe2/operators/experimental/c10/cpu/concat_cpu.cc
caffe2/operators/experimental/c10/cpu/enforce_finite_cpu.cc
caffe2/operators/experimental/c10/cpu/expand_dims_cpu.cc
caffe2/operators/experimental/c10/cpu/fc_cpu.cc
caffe2/operators/experimental/c10/cpu/filler_cpu.cc
caffe2/operators/experimental/c10/cpu/flatten_cpu.cc
caffe2/operators/experimental/c10/cpu/mul_cpu.cc
caffe2/operators/experimental/c10/cpu/relu_cpu.cc
caffe2/operators/experimental/c10/cpu/sigmoid_cpu.cc
caffe2/operators/experimental/c10/cpu/sigmoid_cross_entropy_with_logits_cpu.cc
caffe2/operators/experimental/c10/cpu/sparse_lengths_sum_cpu.cc
caffe2/operators/experimental/c10/cpu/stop_gradient_cpu.cc
caffe2/operators/experimental/c10/schemas/add.cc
caffe2/operators/experimental/c10/schemas/add.h
caffe2/operators/experimental/c10/schemas/averaged_loss.cc
caffe2/operators/experimental/c10/schemas/averaged_loss.h
caffe2/operators/experimental/c10/schemas/batch_gather.cc
caffe2/operators/experimental/c10/schemas/batch_gather.h
caffe2/operators/experimental/c10/schemas/batch_matmul.cc
caffe2/operators/experimental/c10/schemas/batch_matmul.h
caffe2/operators/experimental/c10/schemas/cast.cc
caffe2/operators/experimental/c10/schemas/cast.h
caffe2/operators/experimental/c10/schemas/concat.cc
caffe2/operators/experimental/c10/schemas/concat.h
caffe2/operators/experimental/c10/schemas/enforce_finite.cc
caffe2/operators/experimental/c10/schemas/enforce_finite.h
caffe2/operators/experimental/c10/schemas/expand_dims.cc
caffe2/operators/experimental/c10/schemas/expand_dims.h
caffe2/operators/experimental/c10/schemas/fc.cc
caffe2/operators/experimental/c10/schemas/fc.h
caffe2/operators/experimental/c10/schemas/filler.cc
caffe2/operators/experimental/c10/schemas/filler.h
caffe2/operators/experimental/c10/schemas/flatten.cc
caffe2/operators/experimental/c10/schemas/flatten.h
caffe2/operators/experimental/c10/schemas/layer_norm.cc
caffe2/operators/experimental/c10/schemas/layer_norm.h
caffe2/operators/experimental/c10/schemas/mul.cc
caffe2/operators/experimental/c10/schemas/mul.h
caffe2/operators/experimental/c10/schemas/relu.cc
caffe2/operators/experimental/c10/schemas/relu.h
caffe2/operators/experimental/c10/schemas/sigmoid.cc
caffe2/operators/experimental/c10/schemas/sigmoid.h
caffe2/operators/experimental/c10/schemas/sigmoid_cross_entropy_with_logits.cc
caffe2/operators/experimental/c10/schemas/sigmoid_cross_entropy_with_logits.h
caffe2/operators/experimental/c10/schemas/sparse_lengths_sum.cc
caffe2/operators/experimental/c10/schemas/sparse_lengths_sum.h
caffe2/operators/experimental/c10/schemas/stop_gradient.cc
caffe2/operators/experimental/c10/schemas/stop_gradient.h
caffe2/operators/layer_norm_op.cc