From ae3c5c2ab2c87e4091d6b5b2fd5e22e1c48c62b2 Mon Sep 17 00:00:00 2001 From: sdp Date: Thu, 7 Nov 2024 18:10:21 -0800 Subject: [PATCH] Add aten::lerp_ --- yaml/native/native_functions.yaml | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/yaml/native/native_functions.yaml b/yaml/native/native_functions.yaml index 8fabf229a..3c45f39d2 100644 --- a/yaml/native/native_functions.yaml +++ b/yaml/native/native_functions.yaml @@ -4496,6 +4496,12 @@ structured_delegate: lerp.Scalar_out tags: pointwise +- func: lerp_.Scalar(Tensor(a!) self, Tensor end, Scalar weight) -> Tensor(a!) + device_check: NoCheck # TensorIterator + variants: method + structured_delegate: lerp.Scalar_out + tags: pointwise + - func: lerp.Tensor(Tensor self, Tensor end, Tensor weight) -> Tensor device_check: NoCheck # TensorIterator variants: method, function @@ -4510,6 +4516,12 @@ XPU: lerp_Tensor tags: pointwise +- func: lerp_.Tensor(Tensor(a!) self, Tensor end, Tensor weight) -> Tensor(a!) + device_check: NoCheck # TensorIterator + variants: method + structured_delegate: lerp.Tensor_out + tags: pointwise + - func: addcdiv.out(Tensor self, Tensor tensor1, Tensor tensor2, *, Scalar value=1, Tensor(a!) out) -> Tensor(a!) structured: True structured_inherits: TensorIteratorBase