Rate this Page
★ ★ ★ ★ ★

get_non_observable_arg_indexes_and_types#

class torch.ao.quantization.fx.utils.get_non_observable_arg_indexes_and_types(node)[source]#

Returns a dict with of non float tensor types as keys and values which correspond to a function to retrieve the list (which takes the node as an argument)

Return type:

dict[type | dtype, Callable[[Node], list[int]]]