Searched refs:tf_fused_batch_norm_op (Results 1 – 1 of 1) sorted by relevance
2589 LogicalResult matchAndRewrite(TF::FusedBatchNormOp tf_fused_batch_norm_op, in matchAndRewrite() argument2592 llvm::to_vector<6>(tf_fused_batch_norm_op.getResultTypes()); in matchAndRewrite()2597 OperationState new_state(tf_fused_batch_norm_op.getLoc(), in matchAndRewrite()2599 tf_fused_batch_norm_op.getOperands(), in matchAndRewrite()2601 tf_fused_batch_norm_op.getAttrs()); in matchAndRewrite()2604 rewriter.replaceOp(tf_fused_batch_norm_op, in matchAndRewrite()