Lines Matching defs:shapedValue
147 OpBuilder &b, Location loc, Value shapedValue,
150 if (llvm::isa<RankedTensorType>(shapedValue.getType())) {
151 tensor = shapedValue;
152 } else if (llvm::isa<MemRefType>(shapedValue.getType())) {
153 tensor = b.create<ToTensorOp>(loc, shapedValue);
154 } else if (llvm::isa<UnrankedTensorType>(shapedValue.getType()) ||
155 llvm::isa<UnrankedMemRefType>(shapedValue.getType())) {
156 return getOwnerOfValue(shapedValue)
167 if (llvm::isa<RankedTensorType>(shapedValue.getType()) &&
168 llvm::isa<OpResult>(shapedValue)) {
171 reifyResultShapes(b, shapedValue.getDefiningOp(), resultDims))) {
174 resultDims[llvm::cast<OpResult>(shapedValue).getResultNumber()];