Home
last modified time | relevance | path

Searched refs:op_module (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/ao/quantization/fx/
D_equalize.py510 op_module = modules[str(node.target)][0] # type: ignore[index]
512 op_module = modules[str(node.target)]
514 op_module
515 ) or custom_module_supports_equalization(op_module)
519 weight = op_module.weight
528 op_module.weight = nn.Parameter(scaled_weight)
536 op_module.weight = nn.Parameter(scaled_weight)
539 bias = op_module.bias
547 op_module.bias = nn.Parameter(scaled_bias)
/external/pytorch/torch/
D_ops.py1247 def _get_packet(qualname, op_module): argument
1253 op.__module__ = op_module