aten/src/ATen/nn_parse.py
def base_declaration(func, thnn_function, backends, inplace=False):
    """Creates the NN function without any buffers in it's signature"""
    name, params = re.match(NAME_PARAM_REGEX, func['name']).groups()
    if inplace:
        name += '_'
    params = params.split(', ')
    arguments = [argument_to_declaration(a, func) for a in params]
    if not inplace:
        arguments += output_arguments(thnn_function)
    buffers = [argument_to_declaration('Tensor ' + buf)
               for buf in func.get('buffers', [])]

    return function_info(name, arguments, None, buffers, backends, inplace, func.get('scalar_check'))
Creates the NN function without any buffers in it's signature
Creates the NN function without any buffers in it's signature
Creates the NN function without any buffers in it ' s signature
def base_declaration(func, thnn_function, backends, inplace=False):
    name, params = re.match(NAME_PARAM_REGEX, func['name']).groups()
    if inplace:
        name += '_'
    params = params.split(', ')
    arguments = [argument_to_declaration(a, func) for a in params]
    if not inplace:
        arguments += output_arguments(thnn_function)
    buffers = [argument_to_declaration('Tensor ' + buf)
               for buf in func.get('buffers', [])]
    return function_info(name, arguments, None, buffers, backends, inplace, func.get('scalar_check'))
def base_declaration ( func , thnn_function , backends , inplace = False ) : name , params = re . match ( NAME_PARAM_REGEX , func [ 'name' ] ) . groups ( ) if inplace : name += '_' params = params . split ( ', ' ) arguments = [ argument_to_declaration ( a , func ) for a in params ] if not inplace : arguments += output_arguments ( thnn_function ) buffers = [ argument_to_declaration ( 'Tensor ' + buf ) for buf in func . get ( 'buffers' , [ ] ) ] return function_info ( name , arguments , None , buffers , backends , inplace , func . get ( 'scalar_check' ) )
Creates the NN function without any buffers in it's signature
Creates the NN function without any buffers in it ' s signature
"""Creates the NN function without any buffers in it's signature"""
func
thnn_function
backends
inplace