Lines Matching defs:tdescTy
241 auto tdescTy = getTensorDescType();
242 if (tdescTy.isScattered())
261 auto tdescTy = getTensorDescType();
264 if (tdescTy.getRank() > 2)
267 if (tdescTy.isScattered())
282 auto array_len = tdescTy.getArrayLength();
283 auto tdescShape = getShapeOf(tdescTy);
291 return t >= 0 && t < tdescTy.getRank();
301 if (tdescTy.getRank() == 2) {
317 auto sgMap = tdescTy.getSGMapAttr();
402 auto tdescTy = getTensorDescType();
408 if (!tdescTy.isScattered())
416 auto tdescMemorySpace = static_cast<unsigned>(tdescTy.getMemorySpace());
422 auto chunkSize = tdescTy.getChunkSize();
432 auto elemBits = tdescTy.getElementType().getIntOrFloatBitWidth();
445 if (elemBits * tdescTy.getNumElements() > lscConstraints)
453 auto tdescShape = getShapeOf(tdescTy);
465 auto tdescTy = getTensorDescType();
466 if (!tdescTy.isScattered())
485 auto tdescTy = getTensorDescType();
489 if (!tdescTy.isScattered())
501 auto tdescElemTy = tdescTy.getElementType();
509 auto tdescShape = getShapeOf(tdescTy);
514 if (tdescTy.getRank() == 2) {
532 auto tdescTy = getTensorDescType();
533 if (!tdescTy.isScattered())
548 auto tdescShape = getShapeOf(tdescTy);
553 if (tdescTy.getRank() == 2) {
573 auto tdescTy = mlir::dyn_cast<TensorDescType>(tensorDesc.getType());
574 assert(tdescTy && "Expecting the source is a TensorDescType value.");
580 build(builder, state, tdescTy, tensorDesc, offset);