Home
last modified time | relevance | path

Searched refs:resolveCallable (Results 1 – 17 of 17) sorted by relevance

/external/llvm-project/mlir/lib/Analysis/
DCallGraph.cpp78 parentNode->addCallEdge(cg.resolveCallable(call, symbolTable)); in computeCallGraph()
141 CallGraph::resolveCallable(CallOpInterface call, in resolveCallable() function in CallGraph
143 Operation *callable = call.resolveCallable(&symbolTable); in resolveCallable()
/external/llvm-project/mlir/lib/Interfaces/
DCallInterfaces.cpp22 CallOpInterface::resolveCallable(SymbolTableCollection *symbolTable) { in resolveCallable() function in CallOpInterface
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dexecutor_tpuv1_inline_tpu_island.cc66 dyn_cast_or_null<FuncOp>(call_interface.resolveCallable()); in runOnOperation()
Dtpu_identity_pruning.cc55 auto func = dyn_cast_or_null<FuncOp>(call_op.resolveCallable()); in CollectReachableFunctions()
Dtpu_sharding_identification_pass.cc100 FuncOp func = llvm::dyn_cast<FuncOp>(call_op.resolveCallable()); in GetXlaShardingFromArg()
231 FuncOp func = llvm::dyn_cast<FuncOp>(call_op.resolveCallable()); in GetXlaShardingFromRetval()
Doptimize_global_tensors.cc105 if (auto func = dyn_cast<FuncOp>(call.resolveCallable())) { in AnalyzeFunc()
Dresource_device_inference.cc295 auto func = dyn_cast<FuncOp>(call.resolveCallable()); in runOnOperation()
Dresource_op_lifting_cleanup.cc439 FuncOp func = dyn_cast<FuncOp>(call.resolveCallable()); in CleanupAndCanonicalize()
Dcollection_ops_util.cc195 auto callee = dyn_cast<FuncOp>(call.resolveCallable()); in GetElementTypeFromAccess()
Dshape_inference.cc683 FuncOp func = dyn_cast<FuncOp>(call_op.resolveCallable()); in InferShapeForCall()
1352 if (auto func = dyn_cast<FuncOp>(call_op.resolveCallable())) { in PropagateShapeIntoAttachedFunctions()
Dtensor_array_ops_decomposition.cc503 auto callee = dyn_cast<FuncOp>(call.resolveCallable()); in AccessedGradients()
/external/llvm-project/mlir/include/mlir/Analysis/
DCallGraph.h196 CallGraphNode *resolveCallable(CallOpInterface call,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/
Dresource_alias_analysis.cc197 FuncOp func = dyn_cast<FuncOp>(call.resolveCallable()); in BacktrackValue()
333 FuncOp func = dyn_cast<FuncOp>(call.resolveCallable()); in ResourceAliasAnalysisInfo()
/external/llvm-project/mlir/include/mlir/Interfaces/
DCallInterfaces.td57 Operation *resolveCallable(SymbolTableCollection *symbolTable = nullptr);
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_composite_functions_tf.cc180 if (dyn_cast<FuncOp>(op.resolveCallable()) == lstm_func) { in CheckFusableKerasLstm()
/external/llvm-project/mlir/lib/Transforms/
DInliner.cpp339 CallGraphNode *targetNode = cg.resolveCallable(call, symbolTable); in collectCallOps()
DSCCP.cpp431 Operation *symCallable = callOp.resolveCallable(&symbolTable); in initializeSymbolCallables()