xref: /aosp_15_r20/external/pytorch/test/edge/custom_ops.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45)

#include <ATen/Tensor.h>

namespace custom {
namespace native {
at::Tensor& add_3_out(const at::Tensor& a, const at::Tensor& b, const at::Tensor& c, at::Tensor& out) {
    out = a.add(b).add(c);
    return out;
}
}
}