Searched refs:fused_func_op (Results 1 – 1 of 1) sorted by relevance
50 explicit ConvertLSTMCellSimpleToFusedLSTM(mlir::FuncOp fused_func_op) in ConvertLSTMCellSimpleToFusedLSTM() argument51 : fused_func_op_(fused_func_op), in ConvertLSTMCellSimpleToFusedLSTM()53 builder_(fused_func_op.getBody()) {} in ConvertLSTMCellSimpleToFusedLSTM()178 mlir::FuncOp fused_func_op) in ConvertLayerNormalizedLSTMCellSimpleToFusedLSTM() argument179 : ConvertLSTMCellSimpleToFusedLSTM(fused_func_op) {} in ConvertLayerNormalizedLSTMCellSimpleToFusedLSTM()