tf.contrib.training.byte_size_load_fn
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.
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2020-10-01 UTC.
[null,null,["Last updated 2020-10-01 UTC."],[],[]]
View source on GitHub