Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Transforms/
H A DBufferizableOpInterfaceImpl.cpp373 auto fromElementsOp = cast<tensor::FromElementsOp>(op); in bufferize() local
376 fromElementsOp.getResult().cast<OpResult>(), options); in bufferize()
384 auto tensorType = fromElementsOp.getType().cast<RankedTensorType>(); in bufferize()
388 allocateTensorForShapedValue(rewriter, loc, fromElementsOp.getResult(), in bufferize()
399 if (fromElementsOp.getElements().empty()) { in bufferize()
407 loc, fromElementsOp.getElements().front(), buffer); in bufferize()
420 auto elementIt = fromElementsOp.getElements().begin(); in bufferize()
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp430 if (auto fromElementsOp = getTensor().getDefiningOp<FromElementsOp>()) { in fold() local
431 auto tensorType = fromElementsOp.getType().cast<RankedTensorType>(); in fold()
444 if (static_cast<int>(fromElementsOp.getElements().size()) <= flatIndex || in fold()
447 return fromElementsOp.getElements()[flatIndex]; in fold()