View all cloudpickle analysis

How to use the cloudpickle.CloudPickler.save_reduce function in cloudpickle

To help you get started, we’ve selected a few cloudpickle examples, based on popular ways it is used in public projects.

Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.

tensorflow / probability / tensorflow_probability / python / layers / distribution_layer.py View on Github

def save_tensor(cloud_pickler, tensor, name=None):
    val = tf.get_static_value(tensor)
    if val is None:
      raise ValueError('Cannot pickle Tensor -- '
                       'its value is not known statically: {}.'.format(tensor))
    CloudPickler.save_reduce(cloud_pickler, np.array, (val,))