code: def _pickle_tensor_spec(tensor_spec):
      return _unpickle_tensor_spec, (
          tensor_spec.shape.as_list(),
          tensor_spec.dtype.as_numpy_dtype,
      )
  
file_path: transform/tensorflow_transform/pickle_helper.py
imports: import copyreg
  import tensorflow
  from tensorflow_transform import common
  from tensorflow_metadata.proto.v0 import schema_pb2
  from tensorflow_metadata.proto.v0 import statistics_pb2
label_desc: Use image node_id 3 for calling a global function with example usage: _pickle_tensor_spec(tensor_spec) and returns: _unpickle_tensor_spec,  tensor_spec,  tensor_spec