Home
last modified time | relevance | path

Searched refs:graph_op (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dtf_functional_to_executor.cc72 auto graph_op = builder.create<tf_executor::GraphOp>( in runOnFunction() local
74 graph_op.body().push_back(new Block); in runOnFunction()
75 builder.setInsertionPointToEnd(&graph_op.GetBody()); in runOnFunction()
93 for (auto item : llvm::enumerate(graph_op.getResults())) { in runOnFunction()
Dbreakup-islands.cc63 tf_executor::GraphOp graph_op; in runOnFunction() local
66 graph_op = dyn_cast<tf_executor::GraphOp>(func.front().front()); in runOnFunction()
68 if (!graph_op) { in runOnFunction()
80 llvm::make_early_inc_range(llvm::reverse(graph_op.GetBody()))) { in runOnFunction()
93 llvm::make_early_inc_range(llvm::reverse(graph_op.GetBody()))) { in runOnFunction()
Dexport_graphdef.cc487 auto graph_op = llvm::cast<mlir::tf_executor::GraphOp>(block.front()); in Convert() local
492 const int num_data_results = graph_op.getNumResults(); in Convert()
499 for (const auto& it : llvm::enumerate(graph_op.GetFetch().getOperands())) { in Convert()
560 for (Operation& inst : graph_op.GetBody()) { in Convert()
600 for (Operation& inst : graph_op.GetBody()) { in Convert()
608 exporter.GetControlRetNodes(graph_op.GetFetch(), control_ret_nodes)); in Convert()
Dimport_model.cc349 mlir::FuncOp func, mlir::tf_executor::GraphOp graph_op,
1478 mlir::FuncOp func, mlir::tf_executor::GraphOp graph_op, in ConvertFunctionArgAndRets() argument
1589 builder_.setInsertionPointToEnd(&graph_op.body().front()); in ConvertFunctionArgAndRets()
1590 builder_.create<mlir::tf_executor::FetchOp>(graph_op.getLoc(), in ConvertFunctionArgAndRets()
1594 graph_op.getResults()); in ConvertFunctionArgAndRets()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dexecutor_island_coarsening.cc69 Operation* graph_op = island->getParentOp(); in GetOperandCandidateToMergeWith() local
75 DCHECK_EQ(def->getParentOp(), graph_op); in GetOperandCandidateToMergeWith()
80 island.walk([graph_op, &candidate](Operation* op) { in GetOperandCandidateToMergeWith()
83 if (!def || def->getParentOp() != graph_op) continue; in GetOperandCandidateToMergeWith()
98 Operation* graph_op = island->getParentOp(); in GetResultCandidateToMergeWith() local
103 DCHECK_EQ(user->getParentOp(), graph_op); in GetResultCandidateToMergeWith()
108 Block& graph_body = llvm::cast<GraphOp>(graph_op).GetBody(); in GetResultCandidateToMergeWith()
Dreplicate_to_island.cc235 tf_executor::GraphOp graph_op, in CreateIslandsFromReplicate() argument
286 tf_executor::FetchOp fetch = graph_op.GetFetch(); in CreateIslandsFromReplicate()
309 getOperation().walk([&](tf_executor::GraphOp graph_op) { in runOnFunction() argument
310 for (auto island_op : graph_op.getOps<tf_executor::IslandOp>()) { in runOnFunction()
319 auto graph_op = island_op->getParentOfType<tf_executor::GraphOp>(); in runOnFunction() local
322 if (failed(CreateIslandsFromReplicate(tf_dialect, graph_op, island_op, in runOnFunction()
Dparallel_execute_to_islands.cc168 auto graph_op = island_op->getParentOfType<tf_executor::GraphOp>(); in CreateIslandsFromParallelExecute() local
169 tf_executor::FetchOp fetch = graph_op.GetFetch(); in CreateIslandsFromParallelExecute()
185 getFunction().walk([&](tf_executor::GraphOp graph_op) { in runOnFunction() argument
186 for (auto island_op : graph_op.getOps<tf_executor::IslandOp>()) { in runOnFunction()
Dshape_inference.cc930 if (auto graph_op = dyn_cast<tf_executor::GraphOp>(op)) { in InferShapeForNonTFDialectOperation() local
932 graph_op.GetFetch(), graph_op.GetFetch().fetches(), op->getResults()); in InferShapeForNonTFDialectOperation()
/external/tensorflow/tensorflow/python/framework/
Dops.py7021 def get_resource_handle_data(graph_op): argument
7022 assert type(graph_op) == Tensor # pylint: disable=unidiomatic-typecheck
7025 graph_op.graph._c_graph, graph_op._as_tf_output()) # pylint: disable=protected-access