Searched refs:compute_quant_params (Results 1 – 1 of 1) sorted by relevance
239 std::tuple<double, int> compute_quant_params( in compute_quant_params() function1792 compute_quant_params(input, c10::ScalarType::QUInt8); in test_quantized_conv_transpose2d()1799 const auto weight_quant_params = compute_quant_params(weight, w_dtype); in test_quantized_conv_transpose2d()1813 const auto bias_quant_params = compute_quant_params(bias, bias_dtype); in test_quantized_conv_transpose2d()2120 const auto in1_quant_params = compute_quant_params(input1_cpu); in produce_inputs_for_binary_op()2124 const auto in2_quant_params = compute_quant_params(input2_cpu); in produce_inputs_for_binary_op()2141 const auto in2_quant_params = compute_quant_params(input2_cpu); in produce_inputs_for_binary_op()2291 const auto out_quant_params = compute_quant_params(output_cpu); in test_quantized_binary_op()2420 const auto input_quant_params = compute_quant_params(in_cpu, dtype); in test_max_pool2d()2544 const auto in_quant_params = compute_quant_params(input_cpu, in_dtype); in test_quantized_conv2d()[all …]