MCPcopy
hub / github.com/OpenPPL/ppq / ppq_tensor_round

Function ppq_tensor_round

ppq/utils/round.py:97–113  ·  view source on GitHub ↗

reference: https://en.wikipedia.org/wiki/Rounding Args: value (torch.Tensor): [description] policy (RoundingPolicy, optional): [description]. Defaults to RoundingPolicy.ROUND_HALF_EVEN. Raises: ValueError: [description] Returns: torch.Tensor: [

(
    value: torch.Tensor,
    policy:RoundingPolicy=RoundingPolicy.ROUND_HALF_EVEN)

Source from the content-addressed store, hash-verified

95 raise ValueError('Unexpected rounding policy found.')
96
97def ppq_tensor_round(
98 value: torch.Tensor,
99 policy:RoundingPolicy=RoundingPolicy.ROUND_HALF_EVEN) -> torch.Tensor:
100 """
101 reference: https://en.wikipedia.org/wiki/Rounding
102
103 Args:
104 value (torch.Tensor): [description]
105 policy (RoundingPolicy, optional): [description]. Defaults to RoundingPolicy.ROUND_HALF_EVEN.
106
107 Raises:
108 ValueError: [description]
109
110 Returns:
111 torch.Tensor: [description]
112 """
113 return PPQTensorRoundImpl.apply(value, policy)
114
115def ppq_round_to_power_of_2(value: Union[float, int],
116 policy: RoundingPolicy=RoundingPolicy.ROUND_UP) -> float:

Callers 13

__call__Method · 0.90
forwardMethod · 0.90
forwardMethod · 0.90
forwardMethod · 0.90
forwardMethod · 0.90
PPQLinearQuant_toIntFunction · 0.90
insert_quantize_nodeMethod · 0.90
insert_quantize_nodeMethod · 0.90
__TEST_QUANTIZE_LT__Function · 0.90
__TEST_QUANTIZE_LC__Function · 0.90

Calls

no outgoing calls

Tested by 3

__TEST_QUANTIZE_LT__Function · 0.72
__TEST_QUANTIZE_LC__Function · 0.72
ref_grad_funcFunction · 0.72