Searched defs:graph_op (Results 1 – 11 of 11) sorted by relevance
170 auto graph_op = island_op->getParentOfType<tf_executor::GraphOp>(); in CreateIslandsFromParallelExecute() local187 getOperation().walk([&](tf_executor::GraphOp graph_op) { in runOnOperation()
237 tf_executor::GraphOp graph_op, in CreateIslandsFromReplicate()311 getOperation().walk([&](tf_executor::GraphOp graph_op) { in runOnOperation()321 auto graph_op = island_op->getParentOfType<tf_executor::GraphOp>(); in runOnOperation() local
83 auto graph_op = cast<GraphOp>(func.front().front()); in CollectChainResources() local149 auto graph_op = cast<GraphOp>(func.front().front()); in RemoveAllControlOutputs() local207 auto graph_op = cast<GraphOp>(block.front()); in AppendFunctionResults() local268 auto graph_op = cast<GraphOp>(func.front().front()); in ChainResourceOps() local
1927 if (auto graph_op = dyn_cast<tf_executor::GraphOp>(op)) { in InferShapeForNonTFDialectOperation() local
76 auto graph_op = builder.create<tf_executor::GraphOp>( in runOnOperation() local
83 void AddStatefulControlFlowDependencies(tf_executor::GraphOp graph_op) { in AddStatefulControlFlowDependencies()104 tf_executor::GraphOp graph_op; in runOnFunction() local
458 auto graph_op = llvm::cast<mlir::tf_executor::GraphOp>(block.front()); in Convert() local
1602 mlir::func::FuncOp func, mlir::tf_executor::GraphOp graph_op, in ConvertFunctionArgAndRets()
136 GraphOp graph_op; in ValidateModuleForExport() local
271 auto graph_op = builder.create<GraphOp>( in ConvertGraphDef() local
7352 def get_resource_handle_data(graph_op): argument