Searched defs:maskShape (Results 1 – 5 of 5) sorted by relevance
361 auto maskShape = getShapeOf(maskTy); verify() local 402 auto maskShape = getShapeOf(maskTy); verify() local
504 llvm::ArrayRef<int64_t> maskShape = maskTy.getShape(); verifyLogicalReductionOp() local 582 llvm::ArrayRef<int64_t> maskShape = maskTy.getShape(); verify() local 683 llvm::ArrayRef<int64_t> maskShape; verifyArrayAndMaskForReductionOp() local [all...]
962 if (auto maskShape{(*this)(call.arguments().at(1))}) { operator ()() local
395 auto maskShape = maskType.getShape(); getOrCreateMaskFor() local 3231 __anon729377362e02(ArrayRef<int64_t> maskShape, ArrayRef<bool> scalableDims, Operation *opToMask) depthwiseConv() argument
1035 SmallVector<int64_t> maskShape(numVecDims, ShapedType::kDynamic); getExpectedMaskType() local 3993 SmallVector<int64_t, 8> maskShape = invPermMap.compose(vecType.getShape()); inferTransferOpMaskType() local