| /external/llvm-project/mlir/lib/Conversion/GPUToVulkan/ |
| D | ConvertLaunchFuncToVulkanCalls.cpp | 104 bool isVulkanLaunchCallOp(LLVM::CallOp callOp) { in isVulkanLaunchCallOp()
111 bool isCInterfaceVulkanLaunchCallOp(LLVM::CallOp callOp) { in isCInterfaceVulkanLaunchCallOp()
119 void translateVulkanLaunchCall(LLVM::CallOp vulkanLaunchCallOp);
122 void createBindMemRefCalls(LLVM::CallOp vulkanLaunchCallOp,
126 void collectSPIRVAttributes(LLVM::CallOp vulkanLaunchCallOp);
172 getOperation().walk([this](LLVM::CallOp op) { in runOnOperation()
178 getOperation().walk([this](LLVM::CallOp op) { in runOnOperation()
185 LLVM::CallOp vulkanLaunchCallOp) { in collectSPIRVAttributes()
208 LLVM::CallOp cInterfaceVulkanLaunchCallOp, Value vulkanRuntime) { in createBindMemRefCalls()
248 builder.create<LLVM::CallOp>( in createBindMemRefCalls()
[all …]
|
| /external/llvm-project/mlir/lib/Conversion/AsyncToLLVM/ |
| D | AsyncToLLVM.cpp | 237 blockBuilder.create<LLVM::CallOp>(loc, TypeRange(), in addResumeFunction()
311 builder.create<CallOp>(loc, kCreateToken, TokenType::get(ctx)); in setupCoroMachinery()
325 auto coroId = builder.create<LLVM::CallOp>( in setupCoroMachinery()
330 auto coroSize = builder.create<LLVM::CallOp>( in setupCoroMachinery()
334 auto coroAlloc = builder.create<LLVM::CallOp>( in setupCoroMachinery()
339 auto coroHdl = builder.create<LLVM::CallOp>( in setupCoroMachinery()
352 auto coroMem = builder.create<LLVM::CallOp>( in setupCoroMachinery()
357 builder.create<LLVM::CallOp>(loc, TypeRange(), in setupCoroMachinery()
370 builder.create<LLVM::CallOp>(loc, i1, builder.getSymbolRefAttr(kCoroEnd), in setupCoroMachinery()
434 auto coroSuspend = builder.create<LLVM::CallOp>( in addSuspensionPoint()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/utils/ |
| D | utils.h | 49 LogicalResult CopyAllowedUnregisteredAttrs(Operation* src, CallOp dst,
54 LogicalResult CopyNonSymbolRefAttrs(CallOp src, Operation* dst);
58 void PropagateAttrsToOperations(CallOp src, Block::iterator begin,
|
| D | utils.cc | 34 void CollectAllowedAttrs(CallOp src, NamedAttrList* attrs) { in CollectAllowedAttrs()
136 LogicalResult CopyAllowedUnregisteredAttrs(Operation* src, CallOp dst, in CopyAllowedUnregisteredAttrs()
154 LogicalResult CopyNonSymbolRefAttrs(CallOp src, Operation* dst) { in CopyNonSymbolRefAttrs()
165 void PropagateAttrsToOperations(CallOp src, Block::iterator begin, in PropagateAttrsToOperations()
|
| /external/llvm-project/mlir/lib/Dialect/StandardOps/Transforms/ |
| D | FuncConversions.cpp | 18 struct CallOpSignatureConversion : public OpConversionPattern<CallOp> {
19 using OpConversionPattern<CallOp>::OpConversionPattern;
23 matchAndRewrite(CallOp callOp, ArrayRef<Value> operands, in matchAndRewrite()
33 rewriter.replaceOpWithNewOp<CallOp>(callOp, callOp.callee(), in matchAndRewrite()
|
| D | DecomposeCallGraphTypes.cpp | 126 : public DecomposeCallGraphTypesOpConversionPattern<CallOp> {
131 matchAndRewrite(CallOp op, ArrayRef<Value> operands, in matchAndRewrite()
155 CallOp newCallOp = rewriter.create<CallOp>(op.getLoc(), op.getCallee(), in matchAndRewrite()
|
| D | FuncBufferize.cpp | 40 target.addDynamicallyLegalOp<CallOp>( in runOnOperation()
41 [&](CallOp op) { return typeConverter.isLegal(op); }); in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | raise_to_tf.cc | 72 class RewriteTFRCallOp : public OpRewritePattern<CallOp> {
73 using OpRewritePattern<CallOp>::OpRewritePattern;
78 : OpRewritePattern<CallOp>(context), in RewriteTFRCallOp()
82 LogicalResult matchAndRewrite(CallOp call_op,
101 PatternRewriter& rewriter, TFRFuncOp signature, CallOp call_op,
113 PatternRewriter& rewriter, CallOp call_op,
226 PatternRewriter& rewriter, TFRFuncOp signature, CallOp call_op, in CollectInputsAndAttributes()
359 PatternRewriter& rewriter, CallOp call_op, in CreateAndReplaceOp()
409 CallOp call_op, PatternRewriter& rewriter) const { in matchAndRewrite()
|
| D | decompose.cc | 197 auto new_op = builder.create<CallOp>( in RewriteUnregisteredTFOps()
251 auto walk_result = func.walk([&](CallOp call_op) { in InlineTFRFuncCalls()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | region_control_flow_to_functional.cc | 167 llvm::Optional<CallOp> IsSingleCallRegion(Region& region, in IsSingleCallRegion()
189 CallOp call = dyn_cast<CallOp>(*it++); in IsSingleCallRegion()
214 bool MatchCallArgs(CallOp first, CallOp second, ArgMatcherFn matcher) { in MatchCallArgs()
263 TrivialTransformInfo(llvm::Optional<CallOp> first_call, in TrivialTransformInfo()
264 llvm::Optional<CallOp> second_call, in TrivialTransformInfo()
|
| /external/llvm-project/flang/include/flang/Optimizer/Dialect/ |
| D | FIROpsSupport.h | 27 return isa<fir::CallOp>(op) || isa<fir::DispatchOp>(op) || in isaCall()
28 isa<mlir::CallOp>(op) || isa<mlir::CallIndirectOp>(op); in isaCall()
|
| /external/llvm-project/mlir/lib/Transforms/ |
| D | NormalizeMemRefs.cpp | 129 funcOp.walk([&](CallOp callOp) { in setCalleesAndCallersNonNormalizable()
165 .walk([&](CallOp callOp) -> WalkResult { in areMemRefsNormalizable()
254 auto callOp = dyn_cast<CallOp>(userOp); in updateFunctionSignature()
258 Operation *newCallOp = builder.create<CallOp>( in updateFunctionSignature()
395 op->getNumResults() > 0 && !isa<CallOp>(op) && !funcOp.isExternal()) { in normalizeFuncOpMemRefs()
|
| D | BufferResultsToOutParams.cpp | 84 module.walk([&](CallOp op) { in updateCalls()
112 auto newCall = builder.create<CallOp>(op.getLoc(), op.calleeAttr(), in updateCalls()
|
| /external/llvm-project/mlir/test/lib/Transforms/ |
| D | TestDecomposeCallGraphTypes.cpp | 43 target.addDynamicallyLegalOp<CallOp>( in runOnOperation()
44 [&](CallOp op) { return typeConverter.isLegal(op); }); in runOnOperation()
|
| /external/llvm-project/mlir/lib/Conversion/LinalgToStandard/ |
| D | LinalgToStandard.cpp | 112 rewriter.replaceOpWithNewOp<mlir::CallOp>( in matchAndRewrite()
132 rewriter.replaceOpWithNewOp<mlir::CallOp>( in matchAndRewrite()
180 rewriter.replaceOpWithNewOp<mlir::CallOp>( in matchAndRewrite()
|
| /external/llvm-project/flang/lib/Lower/ |
| D | IO.cpp | 148 builder.create<mlir::CallOp>(loc, getIoMsg, args); in genEndIO()
152 auto call = builder.create<mlir::CallOp>(loc, endIoStatement, endArgs); in genEndIO()
260 ok = builder.create<mlir::CallOp>(loc, outputFunc, outputFuncArgs) in genOutputItemList()
341 ok = builder.create<mlir::CallOp>(loc, inputFunc, inputFuncArgs) in genInputItemList()
348 ok = builder.create<mlir::CallOp>(loc, inputFunc, inputFuncArgs) in genInputItemList()
518 return builder.create<mlir::CallOp>(loc, ioFunc, ioArgs).getResult(0); in genIntIOOption()
534 return builder.create<mlir::CallOp>(loc, ioFunc, ioArgs).getResult(0); in genCharIOOption()
556 return builder.create<mlir::CallOp>(loc, ioFunc, ioArgs).getResult(0); in genIOOption()
616 return builder.create<mlir::CallOp>(loc, ioFunc, ioArgs).getResult(0); in genIOOption()
677 return builder.create<mlir::CallOp>(loc, ioFunc, ioArgs).getResult(0); in genIOOption()
[all …]
|
| /external/llvm-project/mlir/examples/toy/Ch6/mlir/ |
| D | LowerToLLVM.cpp | 86 rewriter.create<CallOp>(loc, printfRef, rewriter.getIntegerType(32), in matchAndRewrite()
95 rewriter.create<CallOp>(loc, printfRef, rewriter.getIntegerType(32), in matchAndRewrite()
|
| /external/llvm-project/mlir/examples/toy/Ch7/mlir/ |
| D | LowerToLLVM.cpp | 86 rewriter.create<CallOp>(loc, printfRef, rewriter.getIntegerType(32), in matchAndRewrite()
95 rewriter.create<CallOp>(loc, printfRef, rewriter.getIntegerType(32), in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | trim_functions_tf.cc | 109 auto walk_result = func.walk([&](CallOp op) -> WalkResult { in Verify()
|
| D | legalize_tf_while.cc | 62 auto call = builder.create<CallOp>(while_op.getLoc(), func, new_operands); in RunOnWhile()
|
| D | while_loop_outline.cc | 68 if (!isa<CallOp>(*it)) return false; in IsAlreadyOutlined()
231 auto call = b.create<CallOp>(while_op.getLoc(), func, new_operands); in OutlineWhile()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | tf_framework_legalize_to_llvm.cc | 109 .create<LLVM::CallOp>( in matchAndRewrite()
231 rewriter.replaceOpWithNewOp<LLVM::CallOp>( in matchAndRewrite()
266 rewriter.replaceOpWithNewOp<LLVM::CallOp>( in matchAndRewrite()
|
| D | embed_memref_prints.cc | 56 return b->create<CallOp>(loc, callee_func, arg); in emitCallToPrint()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_communication.cc | 166 if (!isa<mlir::CallOp>(use.getUser())) continue; in GetFunctionsToRewrite()
382 Value RewriteCallOp(OpBuilder& builder, CallOp call, in RewriteCallOp()
389 auto new_call = builder.create<CallOp>( in RewriteCallOp()
777 } else if (auto call = dyn_cast<mlir::CallOp>(curr_op)) { in RewriteFunction()
826 if (auto call = dyn_cast<mlir::CallOp>(op)) in IsFunctionCallWithCommunication()
|
| /external/llvm-project/mlir/include/mlir/Dialect/StandardOps/EDSC/ |
| D | Intrinsics.h | 21 using std_call = OperationBuilder<CallOp>;
|