From 4e0331b50374db877d27be7617a9d2983ade4bc3 Mon Sep 17 00:00:00 2001 From: Vivek Khandelwal Date: Thu, 25 Apr 2024 12:24:06 +0000 Subject: [PATCH] Address PR comments --- lib/Conversion/TorchOnnxToTorch/DefaultDomainQtoZ.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/lib/Conversion/TorchOnnxToTorch/DefaultDomainQtoZ.cpp b/lib/Conversion/TorchOnnxToTorch/DefaultDomainQtoZ.cpp index 6a4175c78985..b55eb6a33ea2 100644 --- a/lib/Conversion/TorchOnnxToTorch/DefaultDomainQtoZ.cpp +++ b/lib/Conversion/TorchOnnxToTorch/DefaultDomainQtoZ.cpp @@ -851,6 +851,8 @@ void mlir::torch::onnx_c::populateDefaultDomainQtoZ( Torch::ValueTensorType resultType; float alpha, gamma; Value operand; + // Refer https://onnx.ai/onnx/operators/onnx__Selu.html for the default + // alpha and gamma values. if (binder.tensorOperand(operand) || binder.f32FloatAttr(alpha, "alpha", 1.67326) || binder.f32FloatAttr(gamma, "gamma", 1.0507) ||