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

Public Member Functions

def __init__ (self, graph_def, nodes, outfile, dump4tb)
 
def add_operand (self, name, type)
 
def dump_for_tensorboard (self)
 
def get_conv2d_params (self, conv2d_scope_name)
 
def get_dense_params (self, dense_scope_name)
 
def dump_complex_conv2d_to_file (self, node, f)
 
def dump_dense_to_file (self, node, f)
 
def dump_simple_conv2d_to_file (self, node, f)
 
def dump_depth2space_to_file (self, node, f)
 
def dump_mirrorpad_to_file (self, node, f)
 
def dump_maximum_to_file (self, node, f)
 
def dump_mathbinary_to_file (self, node, f)
 
def dump_mathunary_to_file (self, node, f)
 
def dump_avg_pool_to_file (self, node, f)
 
def dump_layers_to_file (self, f)
 
def dump_operands_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 in_conv2d_scope (self, name)
 
def in_dense_scope (self, name)
 
def generate_sub_block_op_scope_info (self)
 
def run (self)
 

Static Public Member Functions

def get_scope_name (name)
 

Data Fields

 graph_def
 
 nodes
 
 outfile
 
 dump4tb
 
 layer_number
 
 output_names
 
 name_node_dict
 
 edges
 
 conv_activations
 
 conv_paddings
 
 pool_paddings
 
 converted_nodes
 
 conv2d_scope_names
 
 conv2d_scopename_inputname_dict
 
 dense_scope_names
 
 dense_scopename_inputname_dict
 
 op2code
 
 mathbin2code
 
 mathun2code
 
 mirrorpad_mode
 
 name_operand_dict
 
 id_different_scope_dict
 

Detailed Description

Definition at line 58 of file convert_from_tensorflow.py.

Constructor & Destructor Documentation

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

Definition at line 59 of file convert_from_tensorflow.py.

Member Function Documentation

def convert_from_tensorflow.TFConverter.add_operand (   self,
  name,
  type 
)
def convert_from_tensorflow.TFConverter.dump_for_tensorboard (   self)

Definition at line 104 of file convert_from_tensorflow.py.

Referenced by convert_from_tensorflow.TFConverter.run().

def convert_from_tensorflow.TFConverter.get_conv2d_params (   self,
  conv2d_scope_name 
)
def convert_from_tensorflow.TFConverter.get_dense_params (   self,
  dense_scope_name 
)
def convert_from_tensorflow.TFConverter.dump_complex_conv2d_to_file (   self,
  node,
  f 
)
def convert_from_tensorflow.TFConverter.dump_dense_to_file (   self,
  node,
  f 
)
def convert_from_tensorflow.TFConverter.dump_simple_conv2d_to_file (   self,
  node,
  f 
)
def convert_from_tensorflow.TFConverter.dump_depth2space_to_file (   self,
  node,
  f 
)
def convert_from_tensorflow.TFConverter.dump_mirrorpad_to_file (   self,
  node,
  f 
)
def convert_from_tensorflow.TFConverter.dump_maximum_to_file (   self,
  node,
  f 
)
def convert_from_tensorflow.TFConverter.dump_mathbinary_to_file (   self,
  node,
  f 
)
def convert_from_tensorflow.TFConverter.dump_mathunary_to_file (   self,
  node,
  f 
)
def convert_from_tensorflow.TFConverter.dump_avg_pool_to_file (   self,
  node,
  f 
)
def convert_from_tensorflow.TFConverter.dump_layers_to_file (   self,
  f 
)
def convert_from_tensorflow.TFConverter.dump_operands_to_file (   self,
  f 
)
def convert_from_tensorflow.TFConverter.dump_to_file (   self)

Definition at line 452 of file convert_from_tensorflow.py.

Referenced by convert_from_tensorflow.TFConverter.run().

def convert_from_tensorflow.TFConverter.generate_name_node_dict (   self)

Definition at line 461 of file convert_from_tensorflow.py.

Referenced by convert_from_tensorflow.TFConverter.run().

def convert_from_tensorflow.TFConverter.generate_output_names (   self)

Definition at line 466 of file convert_from_tensorflow.py.

Referenced by convert_from_tensorflow.TFConverter.run().

def convert_from_tensorflow.TFConverter.remove_identity (   self)

Definition at line 477 of file convert_from_tensorflow.py.

Referenced by convert_from_tensorflow.TFConverter.run().

def convert_from_tensorflow.TFConverter.generate_edges (   self)

Definition at line 505 of file convert_from_tensorflow.py.

Referenced by convert_from_tensorflow.TFConverter.run().

def convert_from_tensorflow.TFConverter.get_scope_name (   name)
static

Definition at line 515 of file convert_from_tensorflow.py.

def convert_from_tensorflow.TFConverter.in_conv2d_scope (   self,
  name 
)
def convert_from_tensorflow.TFConverter.in_dense_scope (   self,
  name 
)
def convert_from_tensorflow.TFConverter.generate_sub_block_op_scope_info (   self)

Definition at line 543 of file convert_from_tensorflow.py.

Referenced by convert_from_tensorflow.TFConverter.run().

def convert_from_tensorflow.TFConverter.run (   self)

Definition at line 585 of file convert_from_tensorflow.py.

Field Documentation

convert_from_tensorflow.TFConverter.graph_def
convert_from_tensorflow.TFConverter.nodes
convert_from_tensorflow.TFConverter.outfile
convert_from_tensorflow.TFConverter.dump4tb

Definition at line 63 of file convert_from_tensorflow.py.

Referenced by convert_from_tensorflow.TFConverter.run().

convert_from_tensorflow.TFConverter.layer_number
convert_from_tensorflow.TFConverter.output_names
convert_from_tensorflow.TFConverter.name_node_dict
convert_from_tensorflow.TFConverter.edges
convert_from_tensorflow.TFConverter.conv_activations
convert_from_tensorflow.TFConverter.conv_paddings
convert_from_tensorflow.TFConverter.pool_paddings
convert_from_tensorflow.TFConverter.converted_nodes
convert_from_tensorflow.TFConverter.conv2d_scope_names
convert_from_tensorflow.TFConverter.conv2d_scopename_inputname_dict
convert_from_tensorflow.TFConverter.dense_scope_names
convert_from_tensorflow.TFConverter.dense_scopename_inputname_dict
convert_from_tensorflow.TFConverter.op2code
convert_from_tensorflow.TFConverter.mathbin2code
convert_from_tensorflow.TFConverter.mathun2code
convert_from_tensorflow.TFConverter.mirrorpad_mode
convert_from_tensorflow.TFConverter.name_operand_dict
convert_from_tensorflow.TFConverter.id_different_scope_dict

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