@@ -4490,10 +4490,12 @@ void Function::createOnnxRNN(llvm::StringRef namePrefix, NodeValue X,
44904490 {idx0, idx1 * hiddenSize, 0 }, { \
44914491 idx0 + 1 , (idx1 + 1 ) * hiddenSize, hiddenSize \
44924492 }
4493+ // clang-format off
44934494#define RNN_B_SLICE_RANGE (idx0, idx1 ) \
44944495 {idx0, idx1 * hiddenSize}, { idx0 + 1 , (idx1 + 1 ) * hiddenSize }
44954496#define RNN_H_SLICE_RANGE (idx ) \
44964497 {idx + 0 , 0 , 0 }, { idx + 1 , batchSize, hiddenSize }
4498+ // clang-format on
44974499#define RNN_CREATE_FC (name, LHS, RHS, BIAS ) \
44984500 BIAS ? (Node *)createFullyConnected (name, LHS, RHS, BIAS) \
44994501 : (Node *)createMatMul (name, LHS, RHS)
@@ -4698,10 +4700,12 @@ void Function::createOnnxGRU(llvm::StringRef namePrefix, NodeValue X,
46984700 {idx0, idx1 * hiddenSize, 0 }, { \
46994701 idx0 + 1 , (idx1 + 1 ) * hiddenSize, hiddenSize \
47004702 }
4703+ // clang-format off
47014704#define GRU_B_SLICE_RANGE (idx0, idx1 ) \
47024705 {idx0, idx1 * hiddenSize}, { idx0 + 1 , (idx1 + 1 ) * hiddenSize }
47034706#define GRU_H_SLICE_RANGE (idx ) \
47044707 {idx + 0 , 0 , 0 }, { idx + 1 , batchSize, hiddenSize }
4708+ // clang-format on
47054709#define GRU_CREATE_FC (name, LHS, RHS, BIAS ) \
47064710 BIAS ? (Node *)createFullyConnected (name, LHS, RHS, BIAS) \
47074711 : (Node *)createMatMul (name, LHS, RHS)
@@ -4981,8 +4985,10 @@ void Function::createOnnxLSTM(llvm::StringRef namePrefix, NodeValue X,
49814985 }
49824986#define LSTM_B_SLICE_RANGE (idx0, idx1 ) \
49834987 {idx0, idx1 * hiddenSize}, { idx0 + 1 , (idx1 + 1 ) * hiddenSize }
4988+ // clang-format off
49844989#define LSTM_P_SLICE_RANGE (idx0, idx1 ) \
49854990 {idx0, idx1 * hiddenSize}, { idx0 + 1 , (idx1 + 1 ) * hiddenSize }
4991+ // clang-format on
49864992#define LSTM_CREATE_FC (name, LHS, RHS, BIAS ) \
49874993 BIAS ? (Node *)createFullyConnected (name, LHS, RHS, BIAS) \
49884994 : (Node *)createMatMul (name, LHS, RHS)
0 commit comments