return std::make_unique<ParentPass>();
}

}  // namespace mlir_dump_test

namespace {

using namespace tensorflow::quantization::mlir_dump_test;

class EnableIrPrintingTest : public ::testing::Test {
 protected:
  EnableIrPrintingTest() : env_(tsl::Env::Default()) {
    if (!tsl::io::GetTestUndeclaredOutputsDir(&test_dir_)) {
      test_dir_ = tsl::testing::TmpDir();
    }
  }

  void SetUp() override {