Searched refs:op_module (Results 1 – 2 of 2) sorted by relevance
510 op_module = modules[str(node.target)][0] # type: ignore[index]512 op_module = modules[str(node.target)]514 op_module515 ) or custom_module_supports_equalization(op_module)519 weight = op_module.weight528 op_module.weight = nn.Parameter(scaled_weight)536 op_module.weight = nn.Parameter(scaled_weight)539 bias = op_module.bias547 op_module.bias = nn.Parameter(scaled_bias)
1247 def _get_packet(qualname, op_module): argument1253 op.__module__ = op_module