Beispiel #1
0
 def _as_variant_tensor(self):
     # pylint: disable=protected-access
     return gen_dataset_ops.directed_interleave_dataset(
         self._selector_input._as_variant_tensor(), [
             data_input._as_variant_tensor()
             for data_input in self._data_inputs
         ], **dataset_ops.flat_structure(self))
 def _as_variant_tensor(self):
   # pylint: disable=protected-access
   return gen_dataset_ops.directed_interleave_dataset(
       self._selector_input._as_variant_tensor(),
       [data_input._as_variant_tensor() for data_input in self._data_inputs],
       output_shapes=nest.flatten(
           sparse.as_dense_shapes(self.output_shapes, self.output_classes)),
       output_types=nest.flatten(
           sparse.as_dense_types(self.output_types, self.output_classes)))
Beispiel #3
0
 def _as_variant_tensor(self):
     # pylint: disable=protected-access
     return gen_dataset_ops.directed_interleave_dataset(
         self._selector_input._as_variant_tensor(), [
             data_input._as_variant_tensor()
             for data_input in self._data_inputs
         ],
         output_shapes=nest.flatten(
             sparse.as_dense_shapes(self.output_shapes,
                                    self.output_classes)),
         output_types=nest.flatten(
             sparse.as_dense_types(self.output_types, self.output_classes)))
Beispiel #4
0
 def _as_variant_tensor(self):
   # pylint: disable=protected-access
   return gen_dataset_ops.directed_interleave_dataset(
       self._selector_input._as_variant_tensor(),
       [data_input._as_variant_tensor() for data_input in self._data_inputs],
       **dataset_ops.flat_structure(self))