format

Signed-off-by: chentong319 <[email protected]>
onnx · Sep 5, 2023 · 994e649 · 994e649
1 parent 27afb6c
commit 994e649
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/src/Conversion/ONNXToKrnl/RNN/GRU.cpp b/src/Conversion/ONNXToKrnl/RNN/GRU.cpp
@@ -506,8 +506,7 @@ void calculateState<GruState, GruActivationPack, GruWeightPack, GruBiasPack>(
                 createMath.cast(sequenceUB.getType(), sequenceIV), sequenceUB);
             nextHt = createMath.select(cond,
                 /*should use initialH*/
-                createMath.constant(nextHt.getType(), 0.),
-                nextHt);
+                createMath.constant(nextHt.getType(), 0.), nextHt);
           }
 
           createKrnl.store(nextHt, Ht, indices);
@@ -621,8 +620,7 @@ void calculateState<GruState, GruActivationPack, GruWeightPack, GruBiasPack>(
                 createMath.cast(sequenceUB.getType(), sequenceIV), sequenceUB);
             nextHt = createMath.select(cond,
                 /*should use initialH*/
-                createMath.constant(nextHt.getType(), 0.),
-                nextHt);
+                createMath.constant(nextHt.getType(), 0.), nextHt);
           }
 
           createKrnl.store(nextHt, Ht, indices);

diff --git a/src/Conversion/ONNXToKrnl/RNN/LSTM.cpp b/src/Conversion/ONNXToKrnl/RNN/LSTM.cpp
@@ -443,7 +443,7 @@ void calculateState<LstmState, LstmActivationPack, LstmWeightPack,
     LstmBiasPack>(ConversionPatternRewriter &rewriter, Location loc, Value Xt,
     LstmState state, LstmActivationPack activationPack,
     LstmWeightPack weightPack, LstmBiasPack biasPack, Value sequenceIV,
-    Value directionIV, Value sequenceLens,  bool isForward) {
+    Value directionIV, Value sequenceLens, bool isForward) {
   // Equations for LSTM.
   // it = f(Xt*(Wi^T) + Ht-1*(Ri^T) + Pi (.) Ct-1 + Wbi + Rbi)
   // ft = f(Xt*(Wf^T) + Ht-1*(Rf^T) + Pf (.) Ct-1 + Wbf + Rbf)