/usr/local/lib64/python3.6/site-packages/torch/include/ATen/native
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/ATen/native/Lerp.h (553B)
#pragma once
#include
#include
namespace at {
namespace native {
using lerp_fn_scalar = void (*)(
at::Tensor& ret,
const at::Tensor& self,
const at::Tensor& end,
const Scalar& weight);
using lerp_fn_tensor = void (*)(
at::Tensor& ret,
const at::Tensor& self,
const at::Tensor& end,
const at::Tensor& weights);
DECLARE_DISPATCH(lerp_fn_scalar, lerp_kernel_scalar_weight);
DECLARE_DISPATCH(lerp_fn_tensor, lerp_kernel_tensor_weight);
} // namespace native
} // namespace at