// Optimizes TF computations by fusing subgraphs/nodes onto more efficient
// implementations to decrease the number of operations needed to perform a
// computation.
struct FusedKernelMatcherPass
    : public impl::FusedKernelMatcherPassBase<FusedKernelMatcherPass> {
  void runOnOperation() override;
};

bool IsActivationFunction(Operation *op) {
  return isa<EluOp, ReluOp, Relu6Op>(op);
}

           "Directory containing the model exported as a TensorFlow "
           "SavedModel. If your model is not based on the TensorFlow "
           "SavedModel, use an empty value.">
  ];
}

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