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

◆ __init__()

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

◆ add_operand()

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

◆ dump_for_tensorboard()

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().

◆ get_conv2d_params()

def convert_from_tensorflow.TFConverter.get_conv2d_params (   self,
  conv2d_scope_name 
)

◆ get_dense_params()

def convert_from_tensorflow.TFConverter.get_dense_params (   self,
  dense_scope_name 
)

◆ dump_complex_conv2d_to_file()

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

◆ dump_dense_to_file()

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

◆ dump_simple_conv2d_to_file()

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

◆ dump_depth2space_to_file()

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

◆ dump_mirrorpad_to_file()

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

◆ dump_maximum_to_file()

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

◆ dump_mathbinary_to_file()

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

◆ dump_mathunary_to_file()

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

◆ dump_avg_pool_to_file()

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

◆ dump_layers_to_file()

def convert_from_tensorflow.TFConverter.dump_layers_to_file (   self,
  f 
)

◆ dump_operands_to_file()

def convert_from_tensorflow.TFConverter.dump_operands_to_file (   self,
  f 
)

◆ dump_to_file()

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().

◆ generate_name_node_dict()

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().

◆ generate_output_names()

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().

◆ remove_identity()

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().

◆ generate_edges()

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().

◆ get_scope_name()

def convert_from_tensorflow.TFConverter.get_scope_name (   name)
static

Definition at line 515 of file convert_from_tensorflow.py.

◆ in_conv2d_scope()

def convert_from_tensorflow.TFConverter.in_conv2d_scope (   self,
  name 
)

◆ in_dense_scope()

def convert_from_tensorflow.TFConverter.in_dense_scope (   self,
  name 
)

◆ generate_sub_block_op_scope_info()

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().

◆ run()

def convert_from_tensorflow.TFConverter.run (   self)

Definition at line 585 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

◆ dump4tb

convert_from_tensorflow.TFConverter.dump4tb

Definition at line 63 of file convert_from_tensorflow.py.

Referenced by convert_from_tensorflow.TFConverter.run().

◆ 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

◆ pool_paddings

convert_from_tensorflow.TFConverter.pool_paddings

◆ converted_nodes

convert_from_tensorflow.TFConverter.converted_nodes

◆ conv2d_scope_names

convert_from_tensorflow.TFConverter.conv2d_scope_names

◆ conv2d_scopename_inputname_dict

convert_from_tensorflow.TFConverter.conv2d_scopename_inputname_dict

◆ dense_scope_names

convert_from_tensorflow.TFConverter.dense_scope_names

◆ dense_scopename_inputname_dict

convert_from_tensorflow.TFConverter.dense_scopename_inputname_dict

◆ op2code

convert_from_tensorflow.TFConverter.op2code

◆ mathbin2code

convert_from_tensorflow.TFConverter.mathbin2code

◆ mathun2code

convert_from_tensorflow.TFConverter.mathun2code

◆ mirrorpad_mode

convert_from_tensorflow.TFConverter.mirrorpad_mode

◆ name_operand_dict

convert_from_tensorflow.TFConverter.name_operand_dict

◆ id_different_scope_dict

convert_from_tensorflow.TFConverter.id_different_scope_dict

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