Searched refs:getScales (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_utils.cc | 130 ArrayRef<double> scales = qtype.getScales(); in RescaleQuantizedType()
212 SmallVector<double, 4> scales(qtype.getScales().begin(), in ResetAxisAndBroadcast()
213 qtype.getScales().end()); in ResetAxisAndBroadcast()
396 if ((axis_size != 1 && axis_size != type.getScales().size())) return {}; in GetUniformQuantizedTypeForBias()
399 axis_size = type.getScales().size(); in GetUniformQuantizedTypeForBias()
411 for (auto index_scale : llvm::enumerate(type.getScales())) { in GetUniformQuantizedTypeForBias()
485 std::transform(uniform_type.getScales().begin(), in QuantizeLegacy()
486 uniform_type.getScales().end(), in QuantizeLegacy()
520 scales.insert(scales.end(), uniform_type.getScales().begin(), in QuantizeLegacy()
521 uniform_type.getScales().end()); in QuantizeLegacy()
|
| D | quantization_utils.h | 420 aqtype.getScales(), new_zero_points, aqtype.getQuantizedDimension(), in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Dialect/Quant/IR/ |
| D | TypeDetail.h | 185 ArrayRef<double> getScales() const { return scales; } in getScales() function
195 lhs.getScales() == rhs.getScales() && in genericIsEqual()
242 ArrayRef<double> getScales() const { in getScales() function
|
| D | QuantTypes.cpp | 346 ArrayRef<double> UniformQuantizedPerAxisType::getScales() const { in getScales() function in UniformQuantizedPerAxisType
347 return getImpl()->getScales(); in getScales()
|
| D | TypeParser.cpp | 403 ArrayRef<double> scales = type.getScales(); in printUniformQuantizedPerAxisType()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_utils.cc | 127 weight_per_channel_qtype.getScales().begin(), in buildRescaleOpConvOutput()
128 weight_per_channel_qtype.getScales().end()); in buildRescaleOpConvOutput()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Quant/ |
| D | QuantTypes.h | 345 ArrayRef<double> getScales() const;
|
| D | UniformSupport.h | 177 : scales(uniformType.getScales()), in UniformQuantizedPerAxisValueConverter()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_export.cc | 813 std::vector<float> scales(qtype.getScales().begin(), in BuildTensor()
814 qtype.getScales().end()); in BuildTensor()
|
| D | flatbuffer_import.cc | 885 per_axis.getExpressedType(), per_axis.getScales(), in PostProcessFuncOp()
|