o->remove();
          block.push_back(o);
        }
      }
    }
  });
  UnmarkCompilationOps(getOperation());
}

}  // namespace

std::unique_ptr<OperationPass<func::FuncOp>> CreateMoveTpuCompileToFrontPass() {
  return std::make_unique<MoveTpuCompileToFrontPass>();
}

}  // namespace TF