def ZeroAllConst()

in src/neuron-gatherinfo/clear_params_tfpb.py [0:0]

• 20 lines of code
• 11 McCabe index (conditional complexity)

def ZeroAllConst(graphdef):
  sess = tf.compat.v1.Session(graph=tf.import_graph_def(graphdef))
  const_by_name = {}
  node_by_name = {}
  for node in graphdef.node:
    node_by_name[node.name] = node  
    if node.op == "Const":
      const_by_name[node.name] = node  
    if node.op == "BiasAdd" or node.op == "MatMul" \
            or node.op.startswith("Conv") \
            or node.op.startswith("FusedBatchNorm"):
      for i in node.input:  
        i_node = node_by_name[i]
        if i_node.op == "Const":
          zero_const(i_node)
        if i_node.op == "Identity":
          x_node = node_by_name[i_node.input[0]]
          if x_node.op == "Const":
            zero_const(x_node)
  return graphdef