See `Dataset.map()` for details.
(self,
input_dataset,
map_func,
use_inter_op_parallelism=True,
preserve_cardinality=False,
use_legacy_function=False)
| 3515 | """A `Dataset` that maps a function over elements in its input.""" |
| 3516 | |
| 3517 | def __init__(self, |
| 3518 | input_dataset, |
| 3519 | map_func, |
| 3520 | use_inter_op_parallelism=True, |
| 3521 | preserve_cardinality=False, |
| 3522 | use_legacy_function=False): |
| 3523 | """See `Dataset.map()` for details.""" |
| 3524 | self._input_dataset = input_dataset |
| 3525 | self._use_inter_op_parallelism = use_inter_op_parallelism |
| 3526 | self._preserve_cardinality = preserve_cardinality |
| 3527 | self._map_func = StructuredFunctionWrapper( |
| 3528 | map_func, |
| 3529 | self._transformation_name(), |
| 3530 | dataset=input_dataset, |
| 3531 | use_legacy_function=use_legacy_function) |
| 3532 | variant_tensor = gen_dataset_ops.map_dataset( |
| 3533 | input_dataset._variant_tensor, # pylint: disable=protected-access |
| 3534 | self._map_func.function.captured_inputs, |
| 3535 | f=self._map_func.function, |
| 3536 | use_inter_op_parallelism=self._use_inter_op_parallelism, |
| 3537 | preserve_cardinality=self._preserve_cardinality, |
| 3538 | **self._flat_structure) |
| 3539 | super(MapDataset, self).__init__(input_dataset, variant_tensor) |
| 3540 | |
| 3541 | def _functions(self): |
| 3542 | return [self._map_func] |
nothing calls this directly
no test coverage detected