xref: /aosp_15_r20/external/pytorch/test/cpp_extensions/jit_extension2.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1 #include <torch/extension.h>
2 
3 using namespace at;
4 
exp_add(Tensor x,Tensor y)5 Tensor exp_add(Tensor x, Tensor y) {
6   return x.exp() + y.exp();
7 }
8