Functions | |
def | load_graph (path) |
def | find_nodes_by_ops (graphdef, ops) |
def | map_node_outputs (graphdef) |
def | get_tensor_metadata (tensors) |
def | get_input_metadata (graph) |
def | get_output_metadata (graph, layerwise=False) |
def | get_graph_output_names (graph) |
def | str_from_graph (graph, mode) |
def polygraphy.backend.tf.util.load_graph | ( | path | ) |
Loads a TensorFlow frozen model. Args: path (Union[str, tf.Graph, tf.GraphDef]): A path to the frozen model, or a frozen TensorFlow graph or graphdef. Returns: tf.Graph: The TensorFlow graph
def polygraphy.backend.tf.util.find_nodes_by_ops | ( | graphdef, | |
ops | |||
) |
def polygraphy.backend.tf.util.map_node_outputs | ( | graphdef | ) |
def polygraphy.backend.tf.util.get_tensor_metadata | ( | tensors | ) |
def polygraphy.backend.tf.util.get_input_metadata | ( | graph | ) |
def polygraphy.backend.tf.util.get_output_metadata | ( | graph, | |
layerwise = False |
|||
) |
def polygraphy.backend.tf.util.get_graph_output_names | ( | graph | ) |
def polygraphy.backend.tf.util.str_from_graph | ( | graph, | |
mode | |||
) |