Home
last modified time | relevance | path

Searched refs:CallOpInterface (Results 1 – 25 of 32) sorted by relevance

12

/external/llvm-project/mlir/lib/Analysis/
DCallGraph.cpp73 if (CallOpInterface call = dyn_cast<CallOpInterface>(op)) { in computeCallGraph()
141 CallGraph::resolveCallable(CallOpInterface call, in resolveCallable()
/external/llvm-project/mlir/include/mlir/Interfaces/
DCallInterfaces.td25 def CallOpInterface : OpInterface<"CallOpInterface"> {
65 be a target for a call-like operation (those providing the CallOpInterface
/external/llvm-project/mlir/include/mlir/Transforms/
DInliningUtils.h25 class CallOpInterface; variable
224 LogicalResult inlineCall(InlinerInterface &interface, CallOpInterface call,
/external/llvm-project/mlir/lib/Interfaces/
DCallInterfaces.cpp22 CallOpInterface::resolveCallable(SymbolTableCollection *symbolTable) { in resolveCallable()
/external/llvm-project/mlir/include/mlir/Analysis/
DCallGraph.h26 class CallOpInterface; variable
196 CallGraphNode *resolveCallable(CallOpInterface call,
/external/llvm-project/mlir/lib/Transforms/
DInliner.cpp302 ResolvedCall(CallOpInterface call, CallGraphNode *sourceNode, in ResolvedCall()
305 CallOpInterface call;
331 if (auto call = dyn_cast<CallOpInterface>(op)) { in collectCallOps()
456 CallOpInterface call = it.call; in inlineCallsInSCC()
DSCCP.cpp202 void visitCallOperation(CallOpInterface op);
430 if (auto callOp = dyn_cast<CallOpInterface>(use.getUser())) { in initializeSymbolCallables()
493 if (CallOpInterface call = dyn_cast<CallOpInterface>(op)) in visitOperation()
574 void SCCPSolver::visitCallOperation(CallOpInterface op) { in visitCallOperation()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dshape_inference.cc499 void PropagateConstantToCallee(CallOpInterface call_op, FuncOp func,
504 void PropagateConstantFromCallee(CallOpInterface call_op, FuncOp func,
571 bool InferShapeForCall(CallOpInterface call_op);
682 bool ShapeInference::InferShapeForCall(CallOpInterface call_op) { in InferShapeForCall()
959 if (auto call = dyn_cast<CallOpInterface>(op)) return InferShapeForCall(call); in InferShapeForNonTFDialectOperation()
1055 if (auto call = dyn_cast<CallOpInterface>(op)) return InferShapeForCall(call); in InferShapeForSingleOperation()
1153 return isa<CallOpInterface>(caller) && in PropagateShapeToFunctions()
1210 void ShapeInference::PropagateConstantToCallee(CallOpInterface call_op, in PropagateConstantToCallee()
1238 void ShapeInference::PropagateConstantFromCallee(CallOpInterface call_op, in PropagateConstantFromCallee()
1351 } else if (auto call_op = dyn_cast<CallOpInterface>(op)) { in PropagateShapeIntoAttachedFunctions()
Dexecutor_tpuv1_inline_tpu_island.cc64 auto call_interface = cast<CallOpInterface>(call_op.getOperation()); in runOnOperation()
Dtpu_identity_pruning.cc54 src.walk([&reachable_funcs, &funcs_to_visit](CallOpInterface call_op) { in CollectReachableFunctions()
Dtpu_sharding_identification_pass.cc99 if (auto call_op = llvm::dyn_cast<CallOpInterface>(owner)) { in GetXlaShardingFromArg()
230 if (auto call_op = llvm::dyn_cast_or_null<CallOpInterface>(def)) { in GetXlaShardingFromRetval()
Dconstant_fold.cc113 if (llvm::isa<TF::WhileOp, TF::CaseOp, TF::IfOp, CallOpInterface>(inst)) { in ConstantFoldFallbackHook()
Doptimize_global_tensors.cc104 if (auto call = dyn_cast<CallOpInterface>(op)) { in AnalyzeFunc()
Dresource_device_inference.cc294 } else if (auto call = dyn_cast<CallOpInterface>(op)) { in runOnOperation()
Dcollection_ops_util.cc194 } else if (auto call = llvm::dyn_cast<CallOpInterface>(use.getOwner())) { in GetElementTypeFromAccess()
Dresource_op_lifting_cleanup.cc438 } else if (auto call = dyn_cast<CallOpInterface>(op)) { in CleanupAndCanonicalize()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/
Dresource_alias_analysis.cc196 } else if (auto call = dyn_cast<CallOpInterface>(op)) { in BacktrackValue()
332 } else if (auto call = dyn_cast<CallOpInterface>(op)) { in ResourceAliasAnalysisInfo()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_composite_functions_tf.cc179 auto result = func.walk([&](CallOpInterface op) { in CheckFusableKerasLstm()
/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
Ddecompose.cc264 cast<CallOpInterface>(call_op.getOperation()), in InlineTFRFuncCalls()
/external/llvm-project/mlir/docs/Tutorials/Toy/
DCh-4.md113 `CallOpInterface`.
126 [DeclareOpInterfaceMethods<CallOpInterface>]> {
299 in a similar way to how we added the `CallOpInterface` to the GenericCallOp:
/external/llvm-project/mlir/lib/Transforms/Utils/
DInliningUtils.cpp298 CallOpInterface call, in inlineCall()
/external/llvm-project/mlir/examples/toy/Ch4/include/toy/
DOps.td127 [DeclareOpInterfaceMethods<CallOpInterface>]> {
/external/llvm-project/mlir/examples/toy/Ch5/include/toy/
DOps.td127 [DeclareOpInterfaceMethods<CallOpInterface>]> {
/external/llvm-project/mlir/examples/toy/Ch6/include/toy/
DOps.td127 [DeclareOpInterfaceMethods<CallOpInterface>]> {
/external/llvm-project/mlir/docs/
DInterfaces.md218 * `CallOpInterface` - Used to represent operations like 'call'

12