Searched refs:tensorShape (Results 1 – 3 of 3) sorted by relevance
34 ArrayRef<int64_t> tensorShape = getInput().getType().getShape(); in verify() local 39 bool compatible = tensorShape.size() == 1 && tensorShape[0] <= polyDegree; in verify() 69 ArrayRef<int64_t> tensorShape = getOutput().getType().getShape(); in verify() local 74 bool compatible = tensorShape.size() == 1 && tensorShape[0] == polyDegree; in verify() 157 ArrayRef<int64_t> tensorShape = tensorType.getShape(); in verifyNTTOp() local 158 bool compatible = tensorShape.size() == 1 && tensorShape[0] == polyDegree; in verifyNTTOp()
289 getPositionFromIndex(int64_t index, llvm::ArrayRef<int64_t> tensorShape) { in getPositionFromIndex() argument291 llvm::SmallVector<int64_t> position(tensorShape.size(), 0); in getPositionFromIndex()292 for (int64_t i = tensorShape.size() - 1; i >= 0; --i) { in getPositionFromIndex()293 position[i] = remaining % tensorShape[i]; in getPositionFromIndex()294 remaining /= tensorShape[i]; in getPositionFromIndex()303 llvm::ArrayRef<int64_t> tensorShape) { in getIndexFromPosition() argument308 multiplierTmp *= tensorShape[i]; in getIndexFromPosition()
188 llvm::SmallVector<int64_t> tensorShape(arrTy.getShape()); in tryCreatingGlobal() 189 std::reverse(tensorShape.begin(), tensorShape.end()); in tryCreatingGlobal() 191 mlir::RankedTensorType::get(tensorShape, attributeElementType); in tryCreatingGlobal() local