Searched refs:QuantParams (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
D | quantization_context.h | 33 static bool EmptyParams(QuantParams p) { return p == quant::QuantizedType(); } in EmptyParams() 38 QuantParams params; 61 QuantParams params; 89 bool SetResultParams(Operation *op, int index, QuantParams params) { in SetResultParams() 96 bool SetOperandParams(Operation *op, int index, QuantParams params) { in SetOperandParams() 101 QuantParams GetResultParams(Operation *op, int index) { in GetResultParams() 106 QuantParams GetOperandParams(Operation *op, int index) { in GetOperandParams() 122 QuantParams GetQuantParamsForSameScaleConstraint(Operation *op); 128 LogicalResult PropagateQuantParams(Operation *op, const QuantParams params, 147 bool SetResultParams(Operation *op, int index, QuantParams params); [all …]
|
D | quantization_driver.cc | 48 static bool EmptyParams(QuantParams p) { return p == quant::QuantizedType(); } in EmptyParams() 53 QuantParams params; 76 QuantParams params; 169 QuantParams GetBiasParams(Operation *op, int bias, 176 bool SetResultParams(Operation *op, int index, QuantParams params); 181 bool SetOperandParams(Operation *op, int index, QuantParams params); 189 void QuantizeOpResult(Operation *op, int index, QuantParams params); 191 void QuantizeArg(BlockArgument arg, QuantParams params); 195 void QuantizeValue(Value value, QuantParams params, Location loc); 209 QuantParams GetQuantParamsForSameScaleConstraint(Operation *op); [all …]
|
D | quantization_context.cc | 193 QuantParams QuantizeContext::GetQuantParamsForSameScaleConstraint( in GetQuantParamsForSameScaleConstraint() 257 Operation *op, const QuantParams params, in PropagateQuantParams() 288 QuantParams params = in InitializeState() 289 params_attr.cast<TypeAttr>().getValue().dyn_cast<QuantParams>(); in InitializeState() 330 QuantParams params) { in SetResultParams() 346 QuantParams params) { in SetOperandParams()
|
D | quantization_info.proto | 49 message QuantParams { message 72 repeated QuantParams entries = 1;
|
D | quantization_utils.h | 52 using QuantParams = quant::QuantizedType; variable 54 using QuantParamsForResults = llvm::SmallVector<QuantParams, 4>; 56 std::function<QuantParams(const std::vector<QuantParams>&, bool)>;
|
D | import_quant_stats_pass.cc | 54 using QuantParamsEntry = QuantizationInfo::QuantParams;
|
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
D | default_quant_params.cc | 73 quant::QuantParams quant_params); 79 quant::QuantParams GetDefaultQuantParams(Builder builder); 83 quant::QuantParams GetQuantParamsForBias(Operation *op, int bias, 90 quant::QuantParams default_quant_params_; 127 quant::QuantParams default_params = GetDefaultQuantParams(builder); in runOnFunction() 138 quant::QuantParams bias_params = GetQuantParamsForBias( in runOnFunction() 167 quant::QuantParams quant_params) { in QuantizeValue() 192 quant::QuantParams DefaultQuantParamsPass::GetQuantParamsForBias( in GetQuantParamsForBias() 215 quant::QuantParams DefaultQuantParamsPass::GetDefaultQuantParams( in GetDefaultQuantParams()
|
D | prepare_quantize_helper.h | 500 return [=](const std::vector<quant::QuantParams>& quant_params, in GetUniformQuantizedTypeForBiasWithScale() 501 bool legacy_float_scale) -> quant::QuantParams { in GetUniformQuantizedTypeForBiasWithScale()
|