Searched defs:QuantArgs (Results 1 – 7 of 7) sorted by relevance
56 def rsqrt_table_8bit(in_quantargs: QuantArgs, out_quantargs: QuantArgs):
64 def div_table_8bit(in_quantargs: QuantArgs, out_quantargs: QuantArgs):
68 def sigmoid_table_8bit(in_quantargs: QuantArgs, out_quantargs: QuantArgs):
68 def tanh_table_8bit(in_quantargs: QuantArgs, out_quantargs: QuantArgs):
67 def exp_table_8bit(in_quantargs: QuantArgs, out_quantargs: QuantArgs):
68 def log_table_8bit(in_quantargs: QuantArgs, out_quantargs: QuantArgs):
45 class QuantArgs(NamedTuple): class65 def quantize_value(x, qargs: QuantArgs, dtype=np.int8):73 def dequantize_value(qx, qargs: QuantArgs):107 def all_q_args_equal(q_arg_list: list[QuantArgs]) -> bool: