Searched refs:input_qscheme (Results 1 – 1 of 1) sorted by relevance
98 input_qscheme=st.sampled_from((torch.per_tensor_affine, torch.per_tensor_symmetric)),102 …def test_input_weight_eq_observer(self, ndim, input_qdtype, input_qscheme, weight_qdtype, weight_q… argument124 input_eq_obs = _InputEqualizationObserver(dtype=input_qdtype, qscheme=input_qscheme)154 input_quant_obs = MinMaxObserver(dtype=input_qdtype, qscheme=input_qscheme)164 if input_qscheme == torch.per_tensor_symmetric: