FFmpeg
Public Member Functions | Data Fields
convert_from_tensorflow.TFConverter Class Reference

Public Member Functions

def __init__ (self, graph_def, nodes, outfile)
 
def dump_for_tensorboard (self)
 
def get_conv2d_params (self, node)
 
def dump_conv2d_to_file (self, node, f)
 
def dump_depth2space_to_file (self, node, f)
 
def generate_layer_number (self)
 
def dump_layers_to_file (self, f)
 
def dump_to_file (self)
 
def generate_name_node_dict (self)
 
def generate_output_names (self)
 
def remove_identity (self)
 
def generate_edges (self)
 
def run (self)
 

Data Fields

 graph_def
 
 nodes
 
 outfile
 
 layer_number
 
 output_names
 
 name_node_dict
 
 edges
 
 conv_activations
 
 conv_paddings
 
 converted_nodes
 
 op2code
 

Detailed Description

Definition at line 29 of file convert_from_tensorflow.py.

Constructor & Destructor Documentation

◆ __init__()

def convert_from_tensorflow.TFConverter.__init__ (   self,
  graph_def,
  nodes,
  outfile 
)

Definition at line 30 of file convert_from_tensorflow.py.

Member Function Documentation

◆ dump_for_tensorboard()

def convert_from_tensorflow.TFConverter.dump_for_tensorboard (   self)

Definition at line 44 of file convert_from_tensorflow.py.

◆ get_conv2d_params()

def convert_from_tensorflow.TFConverter.get_conv2d_params (   self,
  node 
)

◆ dump_conv2d_to_file()

def convert_from_tensorflow.TFConverter.dump_conv2d_to_file (   self,
  node,
  f 
)

◆ dump_depth2space_to_file()

def convert_from_tensorflow.TFConverter.dump_depth2space_to_file (   self,
  node,
  f 
)

◆ generate_layer_number()

def convert_from_tensorflow.TFConverter.generate_layer_number (   self)

◆ dump_layers_to_file()

def convert_from_tensorflow.TFConverter.dump_layers_to_file (   self,
  f 
)

◆ dump_to_file()

def convert_from_tensorflow.TFConverter.dump_to_file (   self)

Definition at line 123 of file convert_from_tensorflow.py.

Referenced by convert_from_tensorflow.TFConverter.run().

◆ generate_name_node_dict()

def convert_from_tensorflow.TFConverter.generate_name_node_dict (   self)

Definition at line 130 of file convert_from_tensorflow.py.

Referenced by convert_from_tensorflow.TFConverter.run().

◆ generate_output_names()

def convert_from_tensorflow.TFConverter.generate_output_names (   self)

Definition at line 135 of file convert_from_tensorflow.py.

Referenced by convert_from_tensorflow.TFConverter.run().

◆ remove_identity()

def convert_from_tensorflow.TFConverter.remove_identity (   self)

Definition at line 146 of file convert_from_tensorflow.py.

Referenced by convert_from_tensorflow.TFConverter.run().

◆ generate_edges()

def convert_from_tensorflow.TFConverter.generate_edges (   self)

Definition at line 172 of file convert_from_tensorflow.py.

Referenced by convert_from_tensorflow.TFConverter.run().

◆ run()

def convert_from_tensorflow.TFConverter.run (   self)

Definition at line 181 of file convert_from_tensorflow.py.

Field Documentation

◆ graph_def

convert_from_tensorflow.TFConverter.graph_def

◆ nodes

convert_from_tensorflow.TFConverter.nodes

◆ outfile

convert_from_tensorflow.TFConverter.outfile

◆ layer_number

convert_from_tensorflow.TFConverter.layer_number

◆ output_names

convert_from_tensorflow.TFConverter.output_names

◆ name_node_dict

convert_from_tensorflow.TFConverter.name_node_dict

◆ edges

convert_from_tensorflow.TFConverter.edges

◆ conv_activations

convert_from_tensorflow.TFConverter.conv_activations

◆ conv_paddings

convert_from_tensorflow.TFConverter.conv_paddings

◆ converted_nodes

convert_from_tensorflow.TFConverter.converted_nodes

◆ op2code

convert_from_tensorflow.TFConverter.op2code

The documentation for this class was generated from the following file: