Home
last modified time | relevance | path

Searched refs:quantizationDim (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Quant/Utils/
H A DUniformSupport.cpp65 size_t dimSize = type.getDimSize(quantizationDim); in convert()
80 std::accumulate(std::next(shape.begin(), quantizationDim + 1), in convert()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Quant/
H A DUniformSupport.h186 quantizationDim(uniformType.getQuantizedDimension()) { in UniformQuantizedPerAxisValueConverter()
215 int32_t quantizationDim; variable