mlir::Location loc,
                                         mlir::Operation* op,
                                         llvm::StringRef device) {
  mlir::OpBuilder::InsertPoint insert_point = builder->saveInsertionPoint();

  auto launch = builder->create<mlir::tf_device::LaunchOp>(
      loc, builder->getStringAttr(device), op->getResultTypes());
  launch.getBody().push_back(new mlir::Block);

          py::return_value_policy::reference)
      .def("getContext", &mlir::OpBuilder::getContext,
           py::return_value_policy::reference);

  py::class_<mlir::OpBuilder::InsertPoint>(m, "OpBuilder_InsertionPoint")
      .def("getBlock", &mlir::OpBuilder::InsertPoint::getBlock);