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.
|
View source on GitHub