| /external/llvm-project/mlir/lib/Dialect/StandardOps/Transforms/ |
| D | FuncConversions.cpp | 23 matchAndRewrite(CallOp callOp, ArrayRef<Value> operands, in matchAndRewrite()
27 if (failed(typeConverter->convertTypes(callOp.getResultTypes(), in matchAndRewrite()
33 rewriter.replaceOpWithNewOp<CallOp>(callOp, callOp.callee(), in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Transforms/ |
| D | NormalizeMemRefs.cpp | 129 funcOp.walk([&](CallOp callOp) { in setCalleesAndCallersNonNormalizable() argument
130 StringRef callee = callOp.getCallee(); in setCalleesAndCallersNonNormalizable()
165 .walk([&](CallOp callOp) -> WalkResult { in areMemRefsNormalizable() argument
167 llvm::seq<unsigned>(0, callOp.getNumResults())) { in areMemRefsNormalizable()
168 Value oldMemRef = callOp.getResult(resIndex); in areMemRefsNormalizable()
254 auto callOp = dyn_cast<CallOp>(userOp); in updateFunctionSignature() local
255 if (!callOp) in updateFunctionSignature()
257 StringRef callee = callOp.getCallee(); in updateFunctionSignature()
|
| D | SCCP.cpp | 430 if (auto callOp = dyn_cast<CallOpInterface>(use.getUser())) { in initializeSymbolCallables() local
431 Operation *symCallable = callOp.resolveCallable(&symbolTable); in initializeSymbolCallables()
434 callToSymbolCallable.try_emplace(callOp, symCallable); in initializeSymbolCallables()
438 if (callOp->getNumResults()) in initializeSymbolCallables()
439 callableLatticeIt->second.addSymbolCall(callOp); in initializeSymbolCallables()
|
| D | Inliner.cpp | 88 void dropCallUses(CallGraphNode *userNode, Operation *callOp, CallGraph &cg);
161 void CGUseList::dropCallUses(CallGraphNode *userNode, Operation *callOp, in dropCallUses() argument
172 walkReferencedSymbolNodes(callOp, cg, symbolTable, resolvedRefs, walkFn); in dropCallUses()
|
| /external/llvm-project/mlir/lib/Conversion/GPUToVulkan/ |
| D | ConvertLaunchFuncToVulkanCalls.cpp | 104 bool isVulkanLaunchCallOp(LLVM::CallOp callOp) { in isVulkanLaunchCallOp() argument
105 return (callOp.callee() && callOp.callee().getValue() == kVulkanLaunch && in isVulkanLaunchCallOp()
106 callOp.getNumOperands() >= kVulkanLaunchNumConfigOperands); in isVulkanLaunchCallOp()
111 bool isCInterfaceVulkanLaunchCallOp(LLVM::CallOp callOp) { in isCInterfaceVulkanLaunchCallOp() argument
112 return (callOp.callee() && in isCInterfaceVulkanLaunchCallOp()
113 callOp.callee().getValue() == kCInterfaceVulkanLaunch && in isCInterfaceVulkanLaunchCallOp()
114 callOp.getNumOperands() >= kVulkanLaunchNumConfigOperands); in isCInterfaceVulkanLaunchCallOp()
|
| /external/llvm-project/mlir/lib/Conversion/GPUCommon/ |
| D | OpToFuncCallLowering.h | 66 auto callOp = rewriter.create<LLVM::CallOp>( in matchAndRewrite() local
71 rewriter.replaceOp(op, {callOp.getResult(0)}); in matchAndRewrite()
76 op->getLoc(), operands.front().getType(), callOp.getResult(0)); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Conversion/SPIRVToLLVM/ |
| D | ConvertSPIRVToLLVM.cpp | 803 matchAndRewrite(spirv::FunctionCallOp callOp, ArrayRef<Value> operands, in matchAndRewrite() argument
805 if (callOp.getNumResults() == 0) { in matchAndRewrite()
806 rewriter.replaceOpWithNewOp<LLVM::CallOp>(callOp, llvm::None, operands, in matchAndRewrite()
807 callOp.getAttrs()); in matchAndRewrite()
812 auto dstType = typeConverter.convertType(callOp.getType(0)); in matchAndRewrite()
813 rewriter.replaceOpWithNewOp<LLVM::CallOp>(callOp, dstType, operands, in matchAndRewrite()
814 callOp.getAttrs()); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Conversion/StandardToLLVM/ |
| D | StandardToLLVM.cpp | 2283 matchAndRewrite(CallOpType callOp, ArrayRef<Value> operands, in matchAndRewrite()
2289 unsigned numResults = callOp.getNumResults(); in matchAndRewrite()
2290 auto resultTypes = llvm::to_vector<4>(callOp.getResultTypes()); in matchAndRewrite()
2299 callOp.getLoc(), /*opOperands=*/callOp->getOperands(), operands, in matchAndRewrite()
2302 callOp.getLoc(), packedResult ? TypeRange(packedResult) : TypeRange(), in matchAndRewrite()
2303 promoted, callOp.getAttrs()); in matchAndRewrite()
2315 this->typeConverter->convertType(callOp.getResult(i).getType()); in matchAndRewrite()
2317 callOp.getLoc(), type, newOp->getResult(0), in matchAndRewrite()
2328 rewriter, callOp.getLoc(), resultTypes, results); in matchAndRewrite()
2329 } else if (failed(copyUnrankedDescriptors(rewriter, callOp.getLoc(), in matchAndRewrite()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_patterns.td | 559 foreach callOp = [TF_PartitionedCallOp, TF_StatefulPartitionedCallOp] in {
560 def : Pat<(callOp:$op $args, FlatSymbolRefAttr:$f,
|