Searched defs:graph_op (Results 1 – 9 of 9) sorted by relevance
168 auto graph_op = island_op->getParentOfType<tf_executor::GraphOp>(); in CreateIslandsFromParallelExecute() local185 getFunction().walk([&](tf_executor::GraphOp graph_op) { in runOnFunction()
235 tf_executor::GraphOp graph_op, in CreateIslandsFromReplicate()309 getOperation().walk([&](tf_executor::GraphOp graph_op) { in runOnFunction()319 auto graph_op = island_op->getParentOfType<tf_executor::GraphOp>(); in runOnFunction() local
69 Operation* graph_op = island->getParentOp(); in GetOperandCandidateToMergeWith() local98 Operation* graph_op = island->getParentOp(); in GetResultCandidateToMergeWith() local
930 if (auto graph_op = dyn_cast<tf_executor::GraphOp>(op)) { in InferShapeForNonTFDialectOperation() local
72 auto graph_op = builder.create<tf_executor::GraphOp>( in runOnFunction() local
63 tf_executor::GraphOp graph_op; in runOnFunction() local
487 auto graph_op = llvm::cast<mlir::tf_executor::GraphOp>(block.front()); in Convert() local
1478 mlir::FuncOp func, mlir::tf_executor::GraphOp graph_op, in ConvertFunctionArgAndRets()
7021 def get_resource_handle_data(graph_op): argument