code: def _get_prediction_data(self):
      invalidInputError(
          not self.drop_remainder,
          "sanity check: drop_remainder should be false in this case,"
          " otherwise please report a bug",
      )
      jvalue = callZooFunc(
          "float",
          "createMiniBatchRDDFromTFDataset",
          self.rdd.map(lambda x: x[0]),
          self.init_op_name,
          self.table_init_op,
          self.output_names,
          self.output_types,
          self.shard_index_op_name,
      )
      rdd = jvalue.value().toJavaRDD()
      return rdd
  
file_path: BigDL/python/orca/src/bigdl/orca/data/tf/tf1_data.py
imports: from bigdl.dllib.utils.common import get_node_and_core_number
  from bigdl.dllib.utils.file_utils import callZooFunc
  from bigdl.dllib.feature.common import FeatureSet
  from bigdl.orca.tfpark import TFDataset
  from bigdl.dllib.utils.log4Error import invalidInputError
  from typing import TYPE_CHECKING, Optional, Any
  import logging
label_desc: Use image node_id 2 for calling the TF1Dataset obj's underlying member method code with example usage: obj._get_prediction_data() and returns: rdd