Home
last modified time | relevance | path

Searched refs:BaseMemRefType (Results 1 – 12 of 12) sorted by relevance

/external/llvm-project/mlir/lib/Transforms/
DBufferResultsToOutParams.cpp29 if (resultType.value().isa<BaseMemRefType>()) { in updateFuncOp()
66 if (operand.getType().isa<BaseMemRefType>()) in updateReturnOps()
88 if (result.getType().isa<BaseMemRefType>()) in updateCalls()
96 if (!memref.getType().cast<BaseMemRefType>().hasStaticShape()) { in updateCalls()
DBufferize.cpp23 assert(inputs[0].getType().isa<BaseMemRefType>()); in materializeTensorLoad()
41 addTargetMaterialization([](OpBuilder &builder, BaseMemRefType type, in BufferizeTypeConverter()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dbuffer_reuse_pass.cc110 if (!value.getType().isa<BaseMemRefType>()) continue; in build()
112 assert(alias.getType().isa<BaseMemRefType>() && in build()
132 if (val.getType().isa<BaseMemRefType>()) buffers.push_back(val); in find_buffer_values()
136 if (val.getType().isa<BaseMemRefType>()) buffers.push_back(val); in find_buffer_values()
216 if (!old_buffer.getType().isa<BaseMemRefType>()) continue; in find_reuse_candiates()
282 if (arg.getType().isa<BaseMemRefType>()) buffer_arguments.push_back(arg); in get_buffer_arguments()
Dbufferize_pass.cc67 assert(inputs[0].getType().isa<BaseMemRefType>()); in materializeTensorLoad()
87 addTargetMaterialization([](OpBuilder& builder, BaseMemRefType type, in CustomBufferizeTypeConverter()
Dtensorflow_abi_knowledge_propagation.cc58 if (argument.getType().isa<BaseMemRefType>()) { in runOnFunction()
/external/llvm-project/mlir/include/mlir/IR/
DBuiltinTypes.h519 class BaseMemRefType : public ShapedType {
543 class MemRefType : public Type::TypeBase<MemRefType, BaseMemRefType,
641 : public Type::TypeBase<UnrankedMemRefType, BaseMemRefType,
714 inline bool BaseMemRefType::classof(Type type) { in classof()
/external/llvm-project/mlir/lib/IR/
DBuiltinTypes.cpp383 unsigned BaseMemRefType::getMemorySpace() const { in getMemorySpace()
427 if (!BaseMemRefType::isValidElementType(elementType)) in getImpl()
495 if (!BaseMemRefType::isValidElementType(elementType)) in verifyConstructionInvariants()
/external/llvm-project/mlir/include/mlir/Conversion/StandardToLLVM/
DConvertStandardToLLVM.h30 class BaseMemRefType; variable
207 Type convertMemRefToBarePtr(BaseMemRefType type);
/external/llvm-project/mlir/lib/Dialect/Linalg/Analysis/
DDependenceAnalysis.cpp33 assert(it->getSecond().getType().isa<BaseMemRefType>() && in find()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dhlo_legalize_to_lhlo.cc590 auto isMemRefType = [](Type type) { return type.isa<BaseMemRefType>(); }; in runOnOperation()
/external/llvm-project/mlir/lib/Conversion/StandardToLLVM/
DStandardToLLVM.cpp373 Type LLVMTypeConverter::convertMemRefToBarePtr(BaseMemRefType type) { in convertMemRefToBarePtr()
425 if (auto memrefTy = type.dyn_cast<BaseMemRefType>()) in convertCallingConventionType()
/external/llvm-project/mlir/lib/Dialect/StandardOps/IR/
DOps.cpp2411 auto srcType = op.source().getType().cast<BaseMemRefType>(); in verify()