Searched refs:lstm_func (Results 1 – 1 of 1) sorted by relevance
130 LogicalResult CheckFusableLayerNormalizedLstmCellSimple(FuncOp lstm_func) { in CheckFusableLayerNormalizedLstmCellSimple() argument132 auto input = lstm_func.getArgument(i); in CheckFusableLayerNormalizedLstmCellSimple()135 lstm_func.emitWarning( in CheckFusableLayerNormalizedLstmCellSimple()145 LogicalResult CheckFusableLstmCellSimple(FuncOp lstm_func) { in CheckFusableLstmCellSimple() argument147 auto input = lstm_func.getArgument(i); in CheckFusableLstmCellSimple()150 lstm_func.emitWarning( in CheckFusableLstmCellSimple()176 LogicalResult CheckFusableKerasLstm(FuncOp lstm_func, ModuleOp module) { in CheckFusableKerasLstm() argument178 if (func == lstm_func) continue; in CheckFusableKerasLstm()180 if (dyn_cast<FuncOp>(op.resolveCallable()) == lstm_func) { in CheckFusableKerasLstm()198 auto input = lstm_func.getArgument(i); in CheckFusableKerasLstm()[all …]