Searched refs:PrepareLstmOutputScale (Results 1 – 2 of 2) sorted by relevance
373 patterns_1.insert<PrepareLstmOutputScale<LSTMOp>>(ctx); in runOnFunction()374 patterns_1.insert<PrepareLstmOutputScale<UnidirectionalSequenceLSTMOp>>( in runOnFunction()
134 struct PrepareLstmOutputScale : public OpRewritePattern<SourceOp> { struct136 explicit PrepareLstmOutputScale(MLIRContext* context) in PrepareLstmOutputScale() argument