/*device_ordinal=*/builder.getI64IntegerAttr(0), device_type_attr),
      builder);
}

// Clones an IfRegionOp 'if_region' and attributes and creates then/else regions
// with yield op and an empty block.
mlir::TF::IfRegionOp CloneEmptyIfWithPredicate(mlir::TF::IfRegionOp if_region,
                                               OpBuilder& builder) {

                                         vars_initialized,
                                         lifted_partitioned_call_callees))) {
        return failure();
      }
    } else if (isa<TF::IfRegionOp, TF::CaseRegionOp, TF::WhileRegionOp>(op)) {
      for (Region& region : op.getRegions())
        (void)HoistForControlFlow(&region.front(), module, vars_initialized,

    }
  }];
}

def TF_YieldOp : TF_Op<"Yield",
      [Terminator,
       Pure,
       NativeOpTrait<"ReturnLike", [], "", "">,
       ParentOneOf<["CaseRegionOp", "IfRegionOp", "WhileRegionOp", "GeneratorDatasetRegionOp"]>,
       DeclareOpInterfaceMethods<RegionBranchTerminatorOpInterface,
           ["getMutableSuccessorOperands"]>,
      ]> {
  let summary = "Yield operation";