Searched refs:getQuantizedDimension (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_utils.cc | 144 new_scales, qtype.getZeroPoints(), qtype.getQuantizedDimension(), in RescaleQuantizedType()
397 if (quant_dim != -1 && quant_dim != type.getQuantizedDimension()) in GetUniformQuantizedTypeForBias()
400 quant_dim = type.getQuantizedDimension(); in GetUniformQuantizedTypeForBias()
492 uniform_type.getQuantizedDimension(), &quantized_values); in QuantizeLegacy()
578 scales, q_type.getZeroPoints(), q_type.getQuantizedDimension(), in DownCastScale()
|
| D | quantization_utils.h | 420 aqtype.getScales(), new_zero_points, aqtype.getQuantizedDimension(), in matchAndRewrite()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Quant/ |
| D | QuantTypes.h | 360 int32_t getQuantizedDimension() const;
|
| D | UniformSupport.h | 183 quantizationDim(uniformType.getQuantizedDimension()) { in UniformQuantizedPerAxisValueConverter()
|
| /external/llvm-project/mlir/lib/Dialect/Quant/IR/ |
| D | TypeParser.cpp | 399 out << type.getQuantizedDimension(); in printUniformQuantizedPerAxisType()
|
| D | QuantTypes.cpp | 354 int32_t UniformQuantizedPerAxisType::getQuantizedDimension() const { in getQuantizedDimension() function in UniformQuantizedPerAxisType
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 886 per_axis.getZeroPoints(), per_axis.getQuantizedDimension(), in PostProcessFuncOp()
|
| D | flatbuffer_export.cc | 819 qtype.getQuantizedDimension()); in BuildTensor()
|