contrib.training.byte_size_load_fn
tf.contrib.training.byte_size_load_fn
tf.contrib.training.byte_size_load_fn
byte_size_load_fn(op)
Defined in tensorflow/contrib/training/python/training/device_setter.py
.
Load function that computes the byte size of a single-output Operation
.
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
: AnOperation
with a single output, typically a "Variable" op.
Returns:
The number of bytes in the output Tensor
.
Raises:
-
ValueError
: ifop
does not have a single output, or if the shape of the single output is not fully-defined.
© 2017 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/contrib/training/byte_size_load_fn