Searched defs:rankedTensorType (Results 1 – 10 of 10) sorted by relevance
89 if (auto rankedTensorType = dyn_cast<RankedTensorType>(tensorType)) { in getBufferType() local
90 RankedTensorType rankedTensorType = dyn_cast<RankedTensorType>(t); populateIteratorTypes() local
632 auto rankedTensorType = llvm::dyn_cast<RankedTensorType>(tensor.getType()); ensureToMemrefOpIsValid() local 779 auto rankedTensorType = llvm::cast<RankedTensorType>(tensorType); getMemRefType() local 800 auto rankedTensorType = llvm::cast<RankedTensorType>(tensorType); getMemRefTypeWithFullyDynamicLayout() local 824 auto rankedTensorType = llvm::cast<RankedTensorType>(tensorType); getMemRefTypeWithStaticIdentityLayout() local [all...]
57 computeTransposedType(RankedTensorType rankedTensorType,ArrayRef<int64_t> transposeVector) computeTransposedType() argument
394 auto rankedTensorType = cast<RankedTensorType>(resultType); in matchAndRewrite() local
173 RankedTensorType rankedTensorType = dyn_cast<RankedTensorType>(type); shardType() local
1502 auto rankedTensorType = matchAndRewrite() local
2477 auto rankedTensorType = llvm::cast<RankedTensorType>(tensor.getType()); createCanonicalRankReducingExtractSliceOp() local 2859 auto rankedTensorType = llvm::cast<RankedTensorType>(dest.getType()); createCanonicalRankReducingInsertSliceOp() local
744 auto rankedTensorType = cast<RankedTensorType>(operand.getType()); broadcastDynamicDimension() local
4505 auto rankedTensorType = foldReadInitWrite() local