Exemplos de QuantLinear.quant_output_scale em Python

Exemplo n.º 1

 def op_symbolic_kwargs(cls, module: QuantLinear):
     linear_symbolic_kwargs = {
         'input_scale': module.quant_input_scale(),
         'input_zero_point': cls.quant_input_zero_point(module),
         'int_weight': cls.int_weight(module).t(),
         'weight_scale': module.quant_weight_scale(),
         'weight_zero_point': cls.quant_weight_zero_point(module),
         'output_scale': module.quant_output_scale(),
         'output_zero_point': cls.quant_output_zero_point(module),
         'output_dtype': cls.torch_8b_dtype(module.is_quant_output_signed),
         'out_shape': cls.quant_output_shape(module)
     }
     return linear_symbolic_kwargs

Exemplo n.º 2

 def op_symbolic_kwargs(cls, module: QuantLinear):
     linear_symbolic_kwargs = {
         'input_scale': module.quant_input_scale(),
         'input_zero_point': cls.quant_input_zero_point(module),
         'int_weight': cls.int_weight(module),
         'weight_scale': module.quant_weight_scale(),
         'weight_zero_point': cls.quant_weight_zero_point(module),
         'output_scale': module.quant_output_scale(),
         'output_zero_point': cls.quant_output_zero_point(module),
         'out_shape': cls.quant_output_shape(module),
         'in_features': module.in_features,
         'out_features': module.out_features
     }
     return linear_symbolic_kwargs