Rate this Page
★ ★ ★ ★ ★

node_arg_is_weight#

class torch.ao.quantization.fx.utils.node_arg_is_weight(node, arg)[source]#

Returns if node arg is weight

Return type:

bool

On this page
Show Source
PyTorch Libraries