patterns.add<FuseConv2DBiasAdd, FuseMatMulBiasAdd>(&getContext());

  (void)applyPatternsAndFoldGreedily(func, std::move(patterns));
}

}  // namespace

std::unique_ptr<OperationPass<func::FuncOp>> CreateFusedKernelMatcherPass() {
  return std::make_unique<FusedKernelMatcherPass>();
}

}  // namespace TF

// generally used beyond exporting to runtimes that supports these ops. In the
// future these fusions may be codegen'd automatically.
std::unique_ptr<OperationPass<func::FuncOp>> CreateFusedKernelMatcherPass();

// Creates function pass to select device index/fold tf.DeviceIndex.
std::unique_ptr<OperationPass<func::FuncOp>> CreateDeviceIndexSelectorPass();

  ];
}

def FusedKernelMatcherPass : Pass<"tf-fused-kernel-matcher", "mlir::func::FuncOp"> {
  let summary = "Matches computations corresponding to optimized fused kernels";
  let constructor = "TF::CreateFusedKernelMatcherPass()";
}

def TFDataOptimizationPass : Pass<"tf-data-optimization", "mlir::func::FuncOp"> {
  let summary = "Performs tf.data optimizations";
  let constructor = "TF::CreateTFDataOptimizationPass()";
}