parameters: (layer)
return_statement: return edge_label
docstring: Define edge label based on layer type.
docstring_summary: Define edge label based on layer type.
docstring_tokens: Define edge label based on layer type .
function: def get_edge_label(layer):
      """Define edge label based on layer type.
      """
  
      if layer.type == 'Data':
          edge_label = 'Batch ' + str(layer.data_param.batch_size)
      elif layer.type == 'Convolution' or layer.type == 'Deconvolution':
          edge_label = str(layer.convolution_param.num_output)
      elif layer.type == 'InnerProduct':
          edge_label = str(layer.inner_product_param.num_output)
      else:
          edge_label = '""'
  
      return edge_label
function_tokens: def get_edge_label ( layer ) : if layer . type == 'Data' : edge_label = 'Batch ' + str ( layer . data_param . batch_size ) elif layer . type == 'Convolution' or layer . type == 'Deconvolution' : edge_label = str ( layer . convolution_param . num_output ) elif layer . type == 'InnerProduct' : edge_label = str ( layer . inner_product_param . num_output ) else : edge_label = '""' return edge_label