#define TENSORFLOW_COMPILER_MLIR_LITE_STABLEHLO_ODML_CONVERTER_FOLDERS_H_

namespace mlir::odml {

// Populates the pattern set with all folding patterns. These patterns
// are intended to have precedence over any other patterns added to the set.
void PopulateFolderPatterns(RewritePatternSet &patternSet);

}  // namespace mlir::odml

namespace mlir {
namespace odml {
// The following 5 different forms of mhlo::iota will be matched:
// 1. IotaOp.
// 2. IotaOp + BroadCastInDim.
// 3. IotaOp + Reshape.
// 4. Constant (folded Iota) + BroadCastInDim.
// 5. Constant (folded result).
// Moreover, the dimensions has to match the iota_dimension.
bool MatchIota(DenseIntElementsAttr dimensions, Value iota);
}  // namespace odml
}  // namespace mlir

    protected String getConfiguredModule() {
        return "idea.module"
    }

    @Override
    protected String projectName(String path) {
        File iml = file(path).listFiles().find { it.name.endsWith("iml") }
        assert iml != null
        return iml.name - ".iml"
    }

    private final IdeaModuleIml iml;
    private final Project project;
    private PathFactory pathFactory;
    private boolean offline;
    private Map<String, Iterable<File>> singleEntryLibraries;

    @Inject
    public IdeaModule(Project project, IdeaModuleIml iml) {
        this.project = project;
        this.iml = iml;

        this.testSources = project.getObjects().fileCollection();

#include <memory>

#include "mlir/Pass/Pass.h"  // from @llvm-project

namespace mlir {
namespace odml {

std::unique_ptr<Pass> CreateRenameEntrypointToMainPass();

}  // namespace odml
}  // namespace mlir

#include <memory>

#include "mlir/Pass/Pass.h"  // from @llvm-project

namespace mlir {
namespace odml {

std::unique_ptr<Pass> CreateSmuggleDisallowedOpsPass();

}  // namespace odml
}  // namespace mlir

#include <memory>

#include "mlir/Pass/Pass.h"  // from @llvm-project

namespace mlir {
namespace odml {

std::unique_ptr<Pass> CreateDropSavedModelSemanticsPass();

}  // namespace odml
}  // namespace mlir

#include "mlir/IR/BuiltinOps.h"  // from @llvm-project
#include "mlir/Pass/Pass.h"  // from @llvm-project

namespace mlir {
namespace odml {

// Creates a pass which transforms TFLite to StableHLO Ops.
std::unique_ptr<OperationPass<func::FuncOp>> CreateTflToStablehloPass();

}  // namespace odml
}  // namespace mlir

#include "mlir/Pass/Pass.h"  // from @llvm-project

namespace mlir::odml {

std::unique_ptr<OperationPass<func::FuncOp>> CreateOutlineCompositesPass();

std::unique_ptr<OperationPass<ModuleOp>> CreateSHLOSimplifyPass();

#define GEN_PASS_REGISTRATION
#include "tensorflow/compiler/mlir/lite/stablehlo/odml_converter/passes.h.inc"

}  // namespace mlir::odml

#define TENSORFLOW_COMPILER_MLIR_LITE_STABLEHLO_TRANSFORMS_CHECK_DIALECTS_PASS_H_

#include <string>

#include "mlir/Pass/Pass.h"  // from @llvm-project

namespace mlir {
namespace odml {

// Creates a pass which checks if there exists allowed dialect ops only or not.
// Based on the list of dialect and op names, it signals failure or not.