Home
last modified time | relevance | path

Searched defs:tensorType (Results 1 – 25 of 29) sorted by relevance

12

/llvm-project/mlir/lib/Dialect/Tosa/IR/
H A DShardingInterfaceImpl.cpp35 auto tensorType = dyn_cast<RankedTensorType>(op->getResult(0).getType()); in getLoopIteratorTypes() local
51 auto tensorType = dyn_cast<RankedTensorType>(op->getResult(0).getType()); in getIndexingMaps() local
/llvm-project/mlir/lib/Dialect/MLProgram/Transforms/
H A DBufferizableOpInterfaceImpl.cpp60 auto tensorType = cast<TensorType>(globalOp.getType()); in bufferize() local
97 auto tensorType = cast<TensorType>(globalLoadOp.getType()); in bufferize() local
127 auto tensorType = cast<TensorType>(globalStoreOp.getValue().getType()); in bufferize() local
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DConvertToDestinationStyle.cpp61 auto tensorType = dyn_cast<RankedTensorType>(tensorSource.getType()); in createMemcpy() local
162 auto tensorType = cast<RankedTensorType>(value.getType()); in reifyOrComputeDynamicSizes() local
196 auto tensorType = cast<RankedTensorType>(value.getType()); global() local
348 RankedTensorType tensorType = rewriteInDestinationPassingStyle() local
390 RankedTensorType tensorType = cast<RankedTensorType>(generateOp.getType()); rewriteInDestinationPassingStyle() local
[all...]
H A DDetensorize.cpp51 bool canBeDetensored(TensorType tensorType) { in canBeDetensored() argument
143 addConversion([](TensorType tensorType) -> Type { in DetensorizeTypeConverter() argument
H A DTransforms.cpp620 permuteShape(RankedTensorType tensorType,ArrayRef<int64_t> permutationVector) permuteShape() argument
640 auto tensorType = permuteShape( transposeOneLinalgOperandAndReplace() local
H A DDropUnitDims.cpp308 if (auto tensorType = dyn_cast<RankedTensorType>(operand.getType())) { collapseValue() local
/llvm-project/mlir/unittests/IR/
H A DShapedTypeTest.cpp79 ShapedType tensorType = TEST() local
197 RankedTensorType tensorType = RankedTensorType::get(shape, f32); TEST() local
[all...]
H A DAttributeTest.cpp84 auto tensorType = RankedTensorType::get({4}, builder.getI1Type()); in TEST() local
/llvm-project/mlir/lib/Dialect/Quant/Utils/
H A DUniformSupport.cpp37 if (auto tensorType = dyn_cast<RankedTensorType>(inputType)) in convert() local
/llvm-project/mlir/include/mlir/Dialect/Bufferization/IR/
H A DUnstructuredControlFlow.h50 auto tensorType = cast<TensorType>(value.getType()); in getBufferType() local
/llvm-project/mlir/lib/Conversion/TensorToSPIRV/
H A DTensorToSPIRV.cpp46 auto tensorType = cast<RankedTensorType>(extractOp.getTensor().getType()); in matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizableOpInterface.cpp161 RankedTensorType tensorType = llvm::cast<RankedTensorType>(tensor.getType()); in allocateTensorForShapedValue() local
365 __anon17a846d30502(TensorType tensorType, Attribute memorySpace, func::FuncOp funcOp, const BufferizationOptions &options) setFunctionBoundaryTypeConversion() argument
642 auto tensorType = llvm::dyn_cast<TensorType>(value.getType()); getBuffer() local
768 auto tensorType = llvm::cast<TensorType>(value.getType()); getMemRefType() local
790 getMemRefTypeWithFullyDynamicLayout(TensorType tensorType,Attribute memorySpace) getMemRefTypeWithFullyDynamicLayout() argument
814 getMemRefTypeWithStaticIdentityLayout(TensorType tensorType,Attribute memorySpace) getMemRefTypeWithStaticIdentityLayout() argument
[all...]
/llvm-project/mlir/lib/Dialect/Polynomial/IR/
H A DPolynomialOps.cpp25 TensorType tensorType = dyn_cast<TensorType>(input.getType()); in build() local
137 RankedTensorType tensorType, in verifyNTTOp() argument
/llvm-project/mlir/lib/Dialect/Tensor/Transforms/
H A DEmptyOpPatterns.cpp84 auto tensorType = RankedTensorType::get(sliceOp.getType().getShape(), in matchAndRewrite() local
H A DRewriteAsConstant.cpp29 auto tensorType = in matchAndRewrite() local
H A DBufferizableOpInterfaceImpl.cpp558 auto tensorType = cast<RankedTensorType>(tensorDestination.getType()); in lowerGenerateLikeOpBody() local
479 auto tensorType = cast<RankedTensorType>(fromElementsOp.getType()); bufferize() local
1024 auto tensorType = cast<RankedTensorType>(tensorAlloc->getType()); bufferize() local
[all...]
/llvm-project/mlir/examples/toy/Ch5/mlir/
H A DLowerToAffineLoops.cpp82 auto tensorType = llvm::cast<RankedTensorType>((*op->result_type_begin())); in lowerOpToLoops() local
164 auto tensorType = llvm::cast<RankedTensorType>(op.getType()); in matchAndRewrite() local
/llvm-project/mlir/examples/toy/Ch6/mlir/
H A DLowerToAffineLoops.cpp82 auto tensorType = llvm::cast<RankedTensorType>((*op->result_type_begin())); in lowerOpToLoops() local
164 auto tensorType = llvm::cast<RankedTensorType>(op.getType()); in matchAndRewrite() local
/llvm-project/mlir/examples/toy/Ch7/mlir/
H A DLowerToAffineLoops.cpp82 auto tensorType = llvm::cast<RankedTensorType>((*op->result_type_begin())); in lowerOpToLoops() local
164 auto tensorType = llvm::cast<RankedTensorType>(op.getType()); in matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Tensor/IR/
H A DValueBoundsOpInterfaceImpl.cpp103 auto tensorType = populateBoundsForIndexValue() local
H A DTensorOps.cpp57 auto tensorType = llvm::cast<RankedTensorType>(value.getType()); getMixedSize() local
67 auto tensorType = llvm::cast<RankedTensorType>(value.getType()); getMixedSizes() local
76 auto tensorType = llvm::dyn_cast<TensorType>(opResult.getType()); getOrCreateDestination() local
538 for (auto tensorType : tensorTypes) inferResultType() local
543 for (auto tensorType : tensorTypes) inferResultType() local
588 for (auto tensorType : inputTypes) { verify() local
599 for (auto tensorType : inputTypes) verify() local
745 auto tensorType = llvm::dyn_cast<RankedTensorType>(getSource().getType()); fold() local
884 auto tensorType = RankedTensorType::get(staticShape, elementType, encoding); build() local
1126 auto tensorType = llvm::cast<RankedTensorType>(getTensor().getType()); verify() local
1149 auto tensorType = llvm::cast<RankedTensorType>(fromElementsOp.getType()); fold() local
[all...]
/llvm-project/mlir/lib/Dialect/Bufferization/Transforms/
H A DFuncBufferizableOpInterfaceImpl.cpp64 tensorType, *options.defaultMemorySpaceFn(tensorType), funcOp, options); in getBufferizedFunctionArgType() local
434 auto tensorType = dyn_cast<TensorType>(returnVal.getType()); bufferize() local
[all...]
H A DBufferize.cpp243 auto tensorType = cast<TensorType>(value.getType()); runOnOperation() local
547 auto tensorType = dyn_cast<TensorType>(bbArg.getType()); bufferizeBlockSignature() local
/llvm-project/mlir/lib/Dialect/EmitC/IR/
H A DEmitC.cpp77 if (auto tensorType = llvm::dyn_cast<TensorType>(type)) { in isSupportedEmitCType() local
1049 Type tensorType = getInitializerTypeForGlobal(getType()); verify() local
/llvm-project/mlir/lib/Dialect/SPIRV/Transforms/
H A DSPIRVConversion.cpp202 if (auto tensorType = dyn_cast<TensorType>(type)) { getTypeNumBytes() local
729 __anonb08dc71c0702(TensorType tensorType) SPIRVTypeConverter() argument

12