Exemplos de round_ste em Python

Exemplo n.º 1

Arquivo: parameter_quant.py Projeto: Pandinosaurus/pytorch-quantization

 def int_weight(self, x: torch.Tensor):
     zero_hw_sentinel = getattr(self, ZERO_HW_SENTINEL_NAME)
     quant_weight, scale, _ = self.tensor_quant(x, zero_hw_sentinel)
     quant_weight = quant_weight / scale
     quant_weight = round_ste(quant_weight)
     quant_weight = quant_weight.int()
     return quant_weight

Exemplo n.º 2

Arquivo: quant_tensor.py Projeto: Pandinosaurus/pytorch-quantization

 def __int__(self):
     return round_ste(self.tensor / self.scale)

Exemplo n.º 3

 def forward(self, x: torch.Tensor):
     return round_ste(x)