| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tfl.cc | 155 input_type.getElementType().isa<mlir::quant::UniformQuantizedType>(); in matchAndRewrite()
157 output_type.getElementType().isa<mlir::quant::UniformQuantizedType>(); in matchAndRewrite()
169 UniformQuantizedType input_qtype = in matchAndRewrite()
171 .dyn_cast<mlir::quant::UniformQuantizedType>(); in matchAndRewrite()
172 UniformQuantizedType output_qtype = in matchAndRewrite()
174 .dyn_cast<mlir::quant::UniformQuantizedType>(); in matchAndRewrite()
212 input_type.getElementType().isa<mlir::quant::UniformQuantizedType>(); in matchAndRewrite()
214 output_type.getElementType().isa<mlir::quant::UniformQuantizedType>(); in matchAndRewrite()
226 UniformQuantizedType input_qtype = in matchAndRewrite()
228 .dyn_cast<mlir::quant::UniformQuantizedType>(); in matchAndRewrite()
[all …]
|
| D | legalize_common.cc | 454 input_lhs_type.getElementType().isa<mlir::quant::UniformQuantizedType>(); in convertMultiplyOp()
456 input_rhs_type.getElementType().isa<mlir::quant::UniformQuantizedType>(); in convertMultiplyOp()
458 output_type.getElementType().isa<mlir::quant::UniformQuantizedType>(); in convertMultiplyOp()
473 .cast<mlir::quant::UniformQuantizedType>(); in convertMultiplyOp()
475 .cast<mlir::quant::UniformQuantizedType>(); in convertMultiplyOp()
477 output_type.getElementType().cast<mlir::quant::UniformQuantizedType>(); in convertMultiplyOp()
637 mlir::quant::UniformQuantizedType lhs_quant_type = in convertConcatV2Op()
639 .dyn_cast_or_null<mlir::quant::UniformQuantizedType>(); in convertConcatV2Op()
640 mlir::quant::UniformQuantizedType rhs_quant_type = in convertConcatV2Op()
642 .dyn_cast_or_null<mlir::quant::UniformQuantizedType>(); in convertConcatV2Op()
[all …]
|
| D | convert_tfl_uint8.cc | 78 .dyn_cast<mlir::quant::UniformQuantizedType>(); in matchAndRewrite()
157 .dyn_cast<mlir::quant::UniformQuantizedType>(); in convert_graph_uint8_tensor()
225 .dyn_cast<mlir::quant::UniformQuantizedType>(); in convert_graph_uint8_tensor()
278 .dyn_cast<mlir::quant::UniformQuantizedType>(); in convert_graph_uint8_tensor()
293 .dyn_cast<mlir::quant::UniformQuantizedType>(); in convert_graph_uint8_tensor()
|
| D | legalize_utils.cc | 84 input_type.getElementType().dyn_cast<mlir::quant::UniformQuantizedType>(); in buildRescaleOpConvOutput()
86 .dyn_cast<mlir::quant::UniformQuantizedType>(); in buildRescaleOpConvOutput()
95 .dyn_cast<mlir::quant::UniformQuantizedType>()) { in buildRescaleOpConvOutput()
177 UniformQuantizedType::get(true, rewriter.getIntegerType(16), in getTosa1DConstTensorTable()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | device_target.cc | 68 if (auto quant = spec.dyn_cast_or_null<UniformQuantizedType>()) { in AppendToSignature()
116 auto in_spec = input_specs[0].dyn_cast<quant::UniformQuantizedType>(); in DecomposeMultiplyAccumulateScale()
118 auto w_spec = input_specs[1].dyn_cast<quant::UniformQuantizedType>(); in DecomposeMultiplyAccumulateScale()
119 auto b_spec = input_specs[2].dyn_cast<quant::UniformQuantizedType>(); in DecomposeMultiplyAccumulateScale()
120 auto o_spec = out_specs[0].dyn_cast<quant::UniformQuantizedType>(); in DecomposeMultiplyAccumulateScale()
165 .dyn_cast<quant::UniformQuantizedType>(); in DecomposeSameScale()
|
| D | quantization_traits.h | 26 using UniformQuantizedType = mlir::quant::UniformQuantizedType; variable
80 return UniformQuantizedType::getChecked( in GetResultQuantizedType()
|
| D | quantization_utils.cc | 401 } else if (!op_type.isa<quant::UniformQuantizedType>()) { in GetUniformQuantizedTypeForBias()
414 } else if (auto type = op_type.dyn_cast<quant::UniformQuantizedType>()) { in GetUniformQuantizedTypeForBias()
434 return quant::UniformQuantizedType::getChecked( in GetUniformQuantizedTypeForBias()
474 if (q_type.isa<UniformQuantizedType>()) { in QuantizeLegacy()
503 if (auto uniform_type = q_type.dyn_cast<UniformQuantizedType>()) { in QuantizeLegacy()
516 if (auto uniform_type = q_type.dyn_cast<UniformQuantizedType>()) { in QuantizeLegacy()
570 if (auto q_type = type.dyn_cast<UniformQuantizedType>()) { in DownCastScale()
571 return UniformQuantizedType::get( in DownCastScale()
704 UniformQuantizedType::getQuantizedElementType(input.getType()); in VerifySameScales()
713 UniformQuantizedType::getQuantizedElementType(output.getType()); in VerifySameScales()
[all …]
|
| D | quantization_utils.h | 404 if (auto uqtype = qtype.template dyn_cast<quant::UniformQuantizedType>()) { in matchAndRewrite()
405 new_qtype = quant::UniformQuantizedType::getChecked( in matchAndRewrite()
578 quant::UniformQuantizedType GetFixedOutputRange(bool is_signed, int bit_width,
|
| D | quantization.td | 76 "UniformQuantizedType", "GetFixedOutputRange",
|
| /external/llvm-project/mlir/include/mlir/Dialect/Quant/ |
| D | QuantOpsBase.td | 62 // An implementation of UniformQuantizedType.
65 CPred<"$_self.isa<UniformQuantizedType>()">,
66 "UniformQuantizedType">;
68 // Predicate for detecting a container or primitive of UniformQuantizedType.
|
| D | QuantizeUtils.h | 18 class UniformQuantizedType; variable
55 UniformQuantizedType quantizedElementType,
|
| D | QuantTypes.h | 254 class UniformQuantizedType
255 : public Type::TypeBase<UniformQuantizedType, QuantizedType,
262 static UniformQuantizedType get(unsigned flags, Type storageType,
269 static UniformQuantizedType
|
| D | FakeQuantSupport.h | 51 UniformQuantizedType fakeQuantAttrsToType(Location loc, unsigned numBits,
|
| D | UniformSupport.h | 61 explicit UniformQuantizedValueConverter(UniformQuantizedType uniformType) in UniformQuantizedValueConverter()
|
| /external/llvm-project/mlir/lib/Dialect/Quant/IR/ |
| D | QuantTypes.cpp | 238 UniformQuantizedType UniformQuantizedType::get(unsigned flags, Type storageType, in get()
247 UniformQuantizedType
248 UniformQuantizedType::getChecked(unsigned flags, Type storageType, in getChecked()
256 LogicalResult UniformQuantizedType::verifyConstructionInvariants( in verifyConstructionInvariants()
284 double UniformQuantizedType::getScale() const { return getImpl()->scale; } in getScale()
286 int64_t UniformQuantizedType::getZeroPoint() const { in getZeroPoint()
|
| D | TypeParser.cpp | 286 return UniformQuantizedType::getChecked(typeFlags, storageType, expressedType, in parseUniformType()
382 static void printUniformQuantizedType(UniformQuantizedType type, in printUniformQuantizedType()
427 else if (auto uniformType = type.dyn_cast<UniformQuantizedType>()) in printType()
|
| D | QuantOps.cpp | 27 addTypes<AnyQuantizedType, CalibratedQuantizedType, UniformQuantizedType, in initialize()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_quantize_helper.h | 293 UniformQuantizedType quant_type = nullptr; in processConstantOp()
306 quant_type = UniformQuantizedType::getChecked( in processConstantOp()
316 .template dyn_cast<quant::UniformQuantizedType>(); in processConstantOp()
368 quant_type = UniformQuantizedType::getChecked( in replaceStatsOp()
463 quant::UniformQuantizedType qtype; in processIntermediates()
473 .template cast<UniformQuantizedType>(); in processIntermediates()
504 .dyn_cast_or_null<UniformQuantizedType>()) { in GetUniformQuantizedTypeForBiasWithScale()
505 return quant::UniformQuantizedType::get( in GetUniformQuantizedTypeForBiasWithScale()
534 !quant_type.template isa<quant::UniformQuantizedType>()) { in GetLstmOpQuantSpec()
540 scale *= quant_type.template dyn_cast<quant::UniformQuantizedType>() in GetLstmOpQuantSpec()
|
| /external/llvm-project/mlir/test/lib/Dialect/Tosa/ |
| D | TosaTestPasses.cpp | 58 outputType.getElementType().dyn_cast<mlir::quant::UniformQuantizedType>(); in matchAndRewrite()
135 inputType.getElementType().dyn_cast<mlir::quant::UniformQuantizedType>(); in matchAndRewrite()
137 weightType.getElementType().dyn_cast<mlir::quant::UniformQuantizedType>(); in matchAndRewrite()
139 outputType.getElementType().dyn_cast<mlir::quant::UniformQuantizedType>(); in matchAndRewrite()
|
| /external/llvm-project/mlir/unittests/Dialect/Quant/ |
| D | QuantizationUtilsTest.cpp | 26 TestUniformQuantizedValueConverter(UniformQuantizedType type) in TestUniformQuantizedValueConverter()
33 UniformQuantizedType qtype;
70 UniformQuantizedType getTestQuantizedType(Type storageType, MLIRContext *ctx) { in getTestQuantizedType()
71 return UniformQuantizedType::get(/*flags=*/false, storageType, in getTestQuantizedType()
|
| /external/llvm-project/mlir/lib/Dialect/Quant/Utils/ |
| D | FakeQuantSupport.cpp | 107 UniformQuantizedType
126 return UniformQuantizedType::getChecked(flags, storageType, expressedType, in fakeQuantAttrsToType()
134 return UniformQuantizedType::getChecked(flags, storageType, expressedType, in fakeQuantAttrsToType()
|
| D | QuantizeUtils.cpp | 100 Attribute realValue, UniformQuantizedType quantizedElementType, in quantizeAttrUniform()
131 quantizedElementType.dyn_cast<UniformQuantizedType>()) { in quantizeAttr()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 175 return mlir::quant::UniformQuantizedType::get( in GetQuantizedType()
867 auto qtype = mlir::quant::UniformQuantizedType::getQuantizedElementType( in PostProcessFuncOp()
889 qtype.dyn_cast<mlir::quant::UniformQuantizedType>()) { in PostProcessFuncOp()
890 new_qtype = mlir::quant::UniformQuantizedType::get( in PostProcessFuncOp()
899 mlir::quant::UniformQuantizedType::castToExpressedType( in PostProcessFuncOp()
|
| /external/llvm-project/mlir/lib/Dialect/Tosa/IR/ |
| D | TosaOps.cpp | 225 .dyn_cast<mlir::quant::UniformQuantizedType>(); in buildMatMulOpWithQuantInfo()
|
| /external/llvm-project/mlir/lib/Dialect/Tosa/Utils/ |
| D | QuantUtils.cpp | 93 ((input_type).getElementType().dyn_cast<quant::UniformQuantizedType>())
|