8e8b81aac14d39288b4c8bbbda86f2cb6a4bb319,tensorflow_transform/tf_utils.py,,deref_tensor_or_op,#Any#,233

Before Change

  if isinstance(tensor_or_op, _RaggedTensorRef):
    // Setting validate=False here because validating entails adding many asserts
    // to the graph every time it"s called.
    return tf.RaggedTensor.from_row_splits(
        values=deref_tensor_or_op(tensor_or_op.values),
        row_splits=deref_tensor_or_op(tensor_or_op.row_splits),
        validate=False)
  return tensor_or_op


def _broadcast_to_x_shape(x, y):

After Change

  if isinstance(tensor_or_op, object_identity.Reference):
    return tensor_or_op.deref()
  if isinstance(tensor_or_op, _CompositeTensorRef):
    return tf.nest.pack_sequence_as(
        structure=tensor_or_op.type_spec,
        flat_sequence=[
            deref_tensor_or_op(component)
            for component in tensor_or_op.list_of_refs
        ],
        expand_composites=True)
  return tensor_or_op


def _broadcast_to_x_shape(x, y):

Instances