Searched defs:PrepareLstmOutputScale (Results 1 – 1 of 1) sorted by relevance
134 struct PrepareLstmOutputScale : public OpRewritePattern<SourceOp> { struct136 explicit PrepareLstmOutputScale(MLIRContext* context) in PrepareLstmOutputScale() argument138 LogicalResult matchAndRewrite(SourceOp op, in matchAndRewrite()170 LogicalResult processRestrictScale(SourceOp op, int input_index, in processRestrictScale()