def run(declarations):
declarations = [d for d in declarations if not exclude(d)]
for declaration in declarations:
common_with_cwrap.set_declaration_defaults(declaration)
common_with_cwrap.enumerate_options_due_to_default(
declaration,
allow_kwarg=False,
type_to_signature=TYPE_FORMAL_GENERIC,
remove_self=True)
common_with_cwrap.sort_by_number_of_options(declaration)
new_options = []
for option in declaration['options']:
set_mode(option)
sanitize_return(option)
process_types_and_backends(option)
add_variants(option)
declaration['options'] = handle_outputs_taken_as_arguments(
declaration['options'])
return declarations
def run(declarations):
declarations = [d for d in declarations if not exclude(d)]
non_extended_methods = set()
for declaration in declarations:
common_with_cwrap.set_declaration_defaults(declaration)
declaration['options'] = [deepcopy(o) for o in declaration['options']]
declaration['options'] = common_with_cwrap.filter_unique_options(
declaration['options'],
allow_kwarg=False,
type_to_signature=TYPE_FORMAL_GENERIC,
remove_self=True)
common_with_cwrap.sort_by_number_of_options(declaration)
discover_zero_dim_tensor_operations(declaration)
discover_sparse_tensor_operations(declaration)
for option in declaration['options']:
set_mode(option)
if option['mode'] != 'native':
sanitize_return(option)
process_types_and_backends(option)
add_variants(option)
if not is_extended_method(option):
non_extended_methods.add(option['api_name'])
declaration['options'] = handle_outputs_taken_as_arguments(
declaration['options'])
# We (very unfortunately) have overloaded virtual methods. Because
# of C++'s rules, we cannot move one overload without doing some
# extra work to make sure that overload in a superclass and an
# overload in a subclass resolve together. I've chosen to resolve
# this problem simply by moving ALL overloads of a method which
# occurs in Tensor to Type. This is why we have to first compute
# which methods *names* go on type, and then move ALL overloads
# of this name to Type.
for declaration in declarations:
for option in declaration['options']:
option['extended_method'] = option[
'api_name'] not in non_extended_methods
return declarations
def run(declarations):
declarations = [d for d in declarations if not exclude(d)]
for declaration in declarations:
common_with_cwrap.set_declaration_defaults(declaration)
declaration['options'] = [deepcopy(o) for o in declaration['options']]
declaration['options'] = common_with_cwrap.filter_unique_options(
declaration['options'],
allow_kwarg=False,
type_to_signature=TYPE_FORMAL_GENERIC,
remove_self=True)
common_with_cwrap.sort_by_number_of_options(declaration)
discover_zero_dim_tensor_operations(declaration)
discover_sparse_tensor_operations(declaration)
for option in declaration['options']:
set_mode(option)
sanitize_return(option)
process_types_and_backends(option)
add_variants(option)
declaration['options'] = handle_outputs_taken_as_arguments(
declaration['options'])
return declarations
def run(declarations):
declarations = [d for d in declarations if not exclude(d)]
for declaration in declarations:
common_with_cwrap.set_declaration_defaults(declaration)
declaration['options'] = [deepcopy(o) for o in declaration['options']]
declaration['options'] = common_with_cwrap.filter_unique_options(
declaration['options'],
allow_kwarg=False,
type_to_signature=TYPE_FORMAL_GENERIC,
remove_self=True)
common_with_cwrap.sort_by_number_of_options(declaration)
discover_zero_dim_tensor_operations(declaration)
discover_sparse_tensor_operations(declaration)
for option in declaration['options']:
set_mode(option)
if option['mode'] != 'native':
sanitize_return(option)
process_types_and_backends(option)
add_variants(option)
declaration['options'] = handle_outputs_taken_as_arguments(
declaration['options'])
return declarations
def run(declarations):
declarations = [d for d in declarations if not exclude(d)]
for declaration in declarations:
common_with_cwrap.set_declaration_defaults(declaration)
common_with_cwrap.enumerate_options_due_to_default(
declaration,
allow_kwarg=False,
type_to_signature=TYPE_FORMAL_GENERIC,
remove_self=True)
common_with_cwrap.sort_by_number_of_options(declaration)
discover_zero_dim_tensor_operations(declaration)
discover_sparse_tensor_operations(declaration)
for option in declaration['options']:
set_mode(option)
sanitize_return(option)
process_types_and_backends(option)
if declaration['name'] == 'multinomial':
print(declaration['name'], option['types'],
option['backend_type_pairs'])
add_variants(option)
declaration['options'] = handle_outputs_taken_as_arguments(
declaration['options'])
return declarations
