Searched defs:rtp (Results 1 – 9 of 9) sorted by relevance
45 auto rtp = cast<ShapedType>(t); convTypes() local 71 auto rtp = cast<RankedTensorType>(type); convVals() local 100 ShapedType rtp = cast<ShapedType>(t); convVals() local [all...]
682 Type rtp = s0.getType(); in matchAndRewrite() local 1077 auto rtp = getRankedTensorType(op.getSrc()); matchAndRewrite() local 1085 auto rtp = getRankedTensorType(op.getResult()); matchAndRewrite() local [all...]
203 const auto rtp = dyn_cast<RankedTensorType>(tensor.getType()); in getNumNonTrivialIdxExpOnSparseLvls() local 1363 if (auto rtp = dyn_cast<RankedTensorType>(operand.getType())) { matchAndRewrite() local
462 SparseInsertGenerator(TensorType rtp,TypeRange retTypes,ValueRange params,bool genCall) SparseInsertGenerator() argument 562 TensorType rtp; global() member in __anonfc1b03f50211::SparseInsertGenerator
50 SparseTensorType(RankedTensorType rtp) in SparseTensorType() argument 378 if (auto rtp = dyn_cast<RankedTensorType>(val.getType())) tryGetSparseTensorType() local [all...]
36 convertSparseTensorType(RankedTensorType rtp, SmallVectorImpl<Type> &fields) { in convertSparseTensorType() argument
170 auto rtp = getRankedTensorType(t); in initialize() local 241 const auto rtp = dyn_cast<RankedTensorType>(tensor.getType()); in initializeLoopEmit() local 306 const auto rtp = dyn_cast<RankedTensorType>(tensor.getType()); in initializeLoopEmit() local 336 auto rtp = dyn_cast<RankedTensorType>(tensors[t].getType()); in initSubSectIterator() local [all...]
416 auto rtp = dyn_cast<RankedTensorType>(type); isZeroRankedTensorOrScalar() local
168 if (auto rtp = dyn_cast<RankedTensorType>(dstTp)) { in genScalarToTensor() local