@@ -4490,10 +4490,12 @@ void Function::createOnnxRNN(llvm::StringRef namePrefix, NodeValue X,
4490
4490
{idx0, idx1 * hiddenSize, 0 }, { \
4491
4491
idx0 + 1 , (idx1 + 1 ) * hiddenSize, hiddenSize \
4492
4492
}
4493
+ // clang-format off
4493
4494
#define RNN_B_SLICE_RANGE (idx0, idx1 ) \
4494
4495
{idx0, idx1 * hiddenSize}, { idx0 + 1 , (idx1 + 1 ) * hiddenSize }
4495
4496
#define RNN_H_SLICE_RANGE (idx ) \
4496
4497
{idx + 0 , 0 , 0 }, { idx + 1 , batchSize, hiddenSize }
4498
+ // clang-format on
4497
4499
#define RNN_CREATE_FC (name, LHS, RHS, BIAS ) \
4498
4500
BIAS ? (Node *)createFullyConnected (name, LHS, RHS, BIAS) \
4499
4501
: (Node *)createMatMul (name, LHS, RHS)
@@ -4698,10 +4700,12 @@ void Function::createOnnxGRU(llvm::StringRef namePrefix, NodeValue X,
4698
4700
{idx0, idx1 * hiddenSize, 0 }, { \
4699
4701
idx0 + 1 , (idx1 + 1 ) * hiddenSize, hiddenSize \
4700
4702
}
4703
+ // clang-format off
4701
4704
#define GRU_B_SLICE_RANGE (idx0, idx1 ) \
4702
4705
{idx0, idx1 * hiddenSize}, { idx0 + 1 , (idx1 + 1 ) * hiddenSize }
4703
4706
#define GRU_H_SLICE_RANGE (idx ) \
4704
4707
{idx + 0 , 0 , 0 }, { idx + 1 , batchSize, hiddenSize }
4708
+ // clang-format on
4705
4709
#define GRU_CREATE_FC (name, LHS, RHS, BIAS ) \
4706
4710
BIAS ? (Node *)createFullyConnected (name, LHS, RHS, BIAS) \
4707
4711
: (Node *)createMatMul (name, LHS, RHS)
@@ -4981,8 +4985,10 @@ void Function::createOnnxLSTM(llvm::StringRef namePrefix, NodeValue X,
4981
4985
}
4982
4986
#define LSTM_B_SLICE_RANGE (idx0, idx1 ) \
4983
4987
{idx0, idx1 * hiddenSize}, { idx0 + 1 , (idx1 + 1 ) * hiddenSize }
4988
+ // clang-format off
4984
4989
#define LSTM_P_SLICE_RANGE (idx0, idx1 ) \
4985
4990
{idx0, idx1 * hiddenSize}, { idx0 + 1 , (idx1 + 1 ) * hiddenSize }
4991
+ // clang-format on
4986
4992
#define LSTM_CREATE_FC (name, LHS, RHS, BIAS ) \
4987
4993
BIAS ? (Node *)createFullyConnected (name, LHS, RHS, BIAS) \
4988
4994
: (Node *)createMatMul (name, LHS, RHS)
0 commit comments