| /external/llvm-project/mlir/lib/Dialect/Quant/Utils/ |
| D | UniformSupport.cpp | 65 size_t dimSize = type.getDimSize(quantizationDim); in convert() local
66 if (dimSize != scales.size()) { in convert()
70 converters.reserve(dimSize); in convert()
71 for (int i = 0, e = dimSize; i != e; ++i) { in convert()
85 return converters[chunkIndex % dimSize].quantizeFloatToInt(old); in convert()
|
| /external/llvm-project/mlir/lib/Analysis/ |
| D | Utils.cpp | 152 int64_t dimSize = memRefType.getDimSize(r); in getConstantBoundingSizeAndShape() local
153 if (ShapedType::isDynamic(dimSize)) in getConstantBoundingSizeAndShape()
155 cstWithShapeBounds.addConstantUpperBound(r, dimSize - 1); in getConstantBoundingSizeAndShape()
173 auto dimSize = memRefType.getDimSize(d); in getConstantBoundingSizeAndShape() local
174 if (dimSize == -1) in getConstantBoundingSizeAndShape()
176 diffConstant = dimSize; in getConstantBoundingSizeAndShape()
475 int64_t dimSize = loadOrStoreOp.getMemRefType().getDimSize(r); in boundCheckLoadOrStoreOp() local
477 if (dimSize == -1) in boundCheckLoadOrStoreOp()
481 ucst.addConstantLowerBound(r, dimSize); in boundCheckLoadOrStoreOp()
|
| /external/llvm-project/mlir/lib/Transforms/ |
| D | PipelineDataTransfer.cpp | 88 for (auto dimSize : oldMemRefType.getShape()) { in doubleBuffer() local
89 if (dimSize == -1) in doubleBuffer()
|
| /external/deqp-deps/glslang/glslang/MachineIndependent/ |
| D | reflection.cpp | 411 … auto dimSize = derefType.isUnsizedArray() ? 0 : derefType.getArraySizes()->getDimSize(0); in blowUpActiveAggregate() local
413 … blockIndex, 0, dimSize, arrayStride, terminalType->getQualifier().storage, false); in blowUpActiveAggregate()
416 … auto dimSize = derefType.isUnsizedArray() ? 0 : derefType.getArraySizes()->getDimSize(0); in blowUpActiveAggregate() local
418 0, dimSize, 0, terminalType->getQualifier().storage, false); in blowUpActiveAggregate()
|
| /external/angle/third_party/vulkan-deps/glslang/src/glslang/MachineIndependent/ |
| D | reflection.cpp | 411 … auto dimSize = derefType.isUnsizedArray() ? 0 : derefType.getArraySizes()->getDimSize(0); in blowUpActiveAggregate() local
413 … blockIndex, 0, dimSize, arrayStride, terminalType->getQualifier().storage, false); in blowUpActiveAggregate()
416 … auto dimSize = derefType.isUnsizedArray() ? 0 : derefType.getArraySizes()->getDimSize(0); in blowUpActiveAggregate() local
418 0, dimSize, 0, terminalType->getQualifier().storage, false); in blowUpActiveAggregate()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 793 auto dimSize = operandType.getDimSize(i); in Verify() local
795 if (dimSize != 1 && dimSize != resultDimSize) { in Verify()
799 i, dimSize, dimIndex, resultDimSize)); in Verify()
881 auto dimSize = operandType.getDimSize(i); in Verify() local
885 if (dimSize != 1 && failed(verifyCompatibleShape(dimSize, resultDimSize))) { in Verify()
889 i, dimSize, dimIndex, resultDimSize)); in Verify()
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorTransforms.cpp | 2016 int64_t dimSize = -1; in lowerParallel() local
2021 dimSize = lhsType.getDimSize(lhsIndex); in lowerParallel()
2025 dimSize = rhsType.getDimSize(rhsIndex); in lowerParallel()
2043 for (int64_t d = 0; d < dimSize; ++d) { in lowerParallel()
2072 int64_t dimSize = lhsType.getDimSize(lhsIndex); in lowerReduction() local
2073 assert(dimSize == rhsType.getDimSize(rhsIndex) && "corrupt shape"); in lowerReduction()
2095 for (int64_t d = 0; d < dimSize; ++d) { in lowerReduction()
|
| /external/llvm-project/flang/lib/Evaluate/ |
| D | fold-integer.cpp | 121 if (auto &dimSize{shape->at(*dim)}) { in UBOUND() local
123 ConvertToType<T>(Expr<ExtentType>{std::move(*dimSize)})); in UBOUND()
|
| /external/llvm-project/mlir/lib/Transforms/Utils/ |
| D | LoopUtils.cpp | 2192 int64_t dimSize = region.memref.getType().cast<MemRefType>().getDimSize(d); in getMultiLevelStrides() local
2193 stride *= dimSize; in getMultiLevelStrides()
2198 if (bufferShape[d] < dimSize && bufferShape[d - 1] > 1) { in getMultiLevelStrides()
2632 auto dimSize = memRefType.getDimSize(d); in getFullMemRefAsRegion() local
2633 assert(dimSize > 0 && "filtered dynamic shapes above"); in getFullMemRefAsRegion()
2635 regionCst->addConstantUpperBound(d, dimSize - 1); in getFullMemRefAsRegion()
|
| /external/llvm-project/mlir/lib/Dialect/StandardOps/IR/ |
| D | Ops.cpp | 322 int64_t dimSize = memrefType.getDimSize(dim); in matchAndRewrite() local
324 if (dimSize != -1) { in matchAndRewrite()
325 newShapeConstants.push_back(dimSize); in matchAndRewrite()
4267 int64_t dimSize = memrefType.getDimSize(dim); in matchAndRewrite() local
4269 if (!ShapedType::isDynamic(dimSize)) { in matchAndRewrite()
4270 newShapeConstants.push_back(dimSize); in matchAndRewrite()
4279 newShapeConstants.push_back(dimSize); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Conversion/StandardToLLVM/ |
| D | StandardToLLVM.cpp | 2991 int64_t dimSize = memRefType.getDimSize(i); in extractSizeOfRankedMemRef() local
2992 return createIndexConstant(rewriter, loc, dimSize); in extractSizeOfRankedMemRef()
|