Rate this Page

return_arg_list#

class torch.ao.quantization.fx.utils.return_arg_list(arg_indices)[source]#

Constructs a function that takes a node as arg and returns the arg_indices that are valid for node.args

Return type:

Callable[[Node], list[int]]