parameters: (function_names)
return_statement: return function_to_args
docstring: Determines argument names for reordered function signatures.
  
    Args:
      function_names: Functions to collect arguments for.
  
    Returns:
      Dictionary mapping function name to its arguments.
docstring_summary: Determines argument names for reordered function signatures.
docstring_tokens: Determines argument names for reordered function signatures .
function: def collect_function_arg_names(function_names):
    """Determines argument names for reordered function signatures.
  
    Args:
      function_names: Functions to collect arguments for.
  
    Returns:
      Dictionary mapping function name to its arguments.
    """
    # Map from reordered function name to its arguments.
    function_to_args = {}
  
    def visit(unused_path, unused_parent, children):
      """Visitor that collects arguments for reordered functions."""
      for child in children:
        _, attr = tf_decorator.unwrap(child[1])
        api_names_v1 = tf_export.get_v1_names(attr)
        api_names_v1 = ['tf.%s' % name for name in api_names_v1]
        matches_function_names = any(
            name in function_names for name in api_names_v1)
        if matches_function_names:
          if tf_inspect.isclass(attr):
            # Get constructor arguments if attr is a class
            arg_list = tf_inspect.getargspec(
                getattr(attr, '__init__'))[0]
            arg_list = arg_list[1:]  # skip 'self' argument
          else:
            # Get function arguments.
            # getargspec returns a tuple of (args, varargs, keywords, defaults)
            # we just look at args.
            arg_list = tf_inspect.getargspec(attr)[0]
          for name in api_names_v1:
            function_to_args[name] = arg_list
  
    visitor = public_api.PublicAPIVisitor(visit)
    visitor.do_not_descend_map['tf'].append('contrib')
    visitor.do_not_descend_map['tf.compat'] = ['v1', 'v2']
    traverse.traverse(tf, visitor)
  
    return function_to_args
function_tokens: def collect_function_arg_names ( function_names ) : # Map from reordered function name to its arguments. function_to_args = { } def visit ( unused_path , unused_parent , children ) : """Visitor that collects arguments for reordered functions.""" for child in children : _ , attr = tf_decorator . unwrap ( child [ 1 ] ) api_names_v1 = tf_export . get_v1_names ( attr ) api_names_v1 = [ 'tf.%s' % name for name in api_names_v1 ] matches_function_names = any ( name in function_names for name in api_names_v1 ) if matches_function_names : if tf_inspect . isclass ( attr ) : # Get constructor arguments if attr is a class arg_list = tf_inspect . getargspec ( getattr ( attr , '__init__' ) ) [ 0 ] arg_list = arg_list [ 1 : ] # skip 'self' argument else : # Get function arguments. # getargspec returns a tuple of (args, varargs, keywords, defaults) # we just look at args. arg_list = tf_inspect . getargspec ( attr ) [ 0 ] for name in api_names_v1 : function_to_args [ name ] = arg_list visitor = public_api . PublicAPIVisitor ( visit ) visitor . do_not_descend_map [ 'tf' ] . append ( 'contrib' ) visitor . do_not_descend_map [ 'tf.compat' ] = [ 'v1' , 'v2' ] traverse . traverse ( tf , visitor ) return function_to_args