View source on GitHub
|
Class TransformerInfo
"Contains information about the result of a transform operation.
__init__
__init__(info)
Constructor.
Args:
info: an instance of Transformer._TmpInfo containing various internal information about the transform operation.
Methods
tf.contrib.graph_editor.TransformerInfo.original
original(
transformed,
missing_fn=None
)
Return the original op/tensor corresponding to the transformed one.
Note that the output of this function mimics the hierarchy
of its input argument transformed.
Given an iterable, it returns a list. Given an operation or a tensor,
it will return an operation or a tensor.
Args:
transformed: the transformed tensor/operation.missing_fn: function handling the case where the counterpart cannot be found. By default, None is returned.
Returns:
the original tensor/operation (or None if no match is found).
tf.contrib.graph_editor.TransformerInfo.transformed
transformed(
original,
missing_fn=None
)
Return the transformed op/tensor corresponding to the original one.
Note that the output of this function mimics the hierarchy
of its input argument original.
Given an iterable, it returns a list. Given an operation or a tensor,
it will return an operation or a tensor.
Args:
original: the original tensor/operation.missing_fn: function handling the case where the counterpart cannot be found. By default, None is returned.
Returns:
the transformed tensor/operation (or None if no match is found).