Home
last modified time | relevance | path

Searched refs:QuantParams (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dquantization_context.h33 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 …]
Dquantization_driver.cc48 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 …]
Dquantization_context.cc193 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()
Dquantization_info.proto49 message QuantParams { message
72 repeated QuantParams entries = 1;
Dquantization_utils.h52 using QuantParams = quant::QuantizedType; variable
54 using QuantParamsForResults = llvm::SmallVector<QuantParams, 4>;
56 std::function<QuantParams(const std::vector<QuantParams>&, bool)>;
Dimport_quant_stats_pass.cc54 using QuantParamsEntry = QuantizationInfo::QuantParams;
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Ddefault_quant_params.cc73 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()
Dprepare_quantize_helper.h500 return [=](const std::vector<quant::QuantParams>& quant_params, in GetUniformQuantizedTypeForBiasWithScale()
501 bool legacy_float_scale) -> quant::QuantParams { in GetUniformQuantizedTypeForBiasWithScale()