TensorFlow 2.0 RC is available Learn more

tf.contrib.training.byte_size_load_fn

View source on GitHub

Load function that computes the byte size of a single-output Operation.

tf.contrib.training.byte_size_load_fn(op)

This is intended to be used with "Variable" ops, which have a single Tensor output with the contents of the variable. However, it can also be used for calculating the size of any op that has a single output.

Intended to be used with GreedyLoadBalancingStrategy.

Args:

  • op: An Operation with a single output, typically a "Variable" op.

Returns:

The number of bytes in the output Tensor.

Raises:

  • ValueError: if op does not have a single output, or if the shape of the single output is not fully-defined.