Searched refs:GetFetch (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | graph_pruning.cc | 145 reachable_ops.insert(graph.GetFetch()); in PruneGraph()
146 VisitOpOperands(graph, graph.GetFetch(), &reachable_ops, &ops_to_visit); in PruneGraph()
|
| D | parallel_execute_to_islands.cc | 169 tf_executor::FetchOp fetch = graph_op.GetFetch(); in CreateIslandsFromParallelExecute()
|
| D | tpu_device_propagation.cc | 240 PropagateDevicesToResults(func, graph.GetFetch(), value_to_device); in runOnFunction()
|
| D | replicate_to_island.cc | 286 tf_executor::FetchOp fetch = graph_op.GetFetch(); in CreateIslandsFromReplicate()
|
| D | executor_island_coarsening.cc | 323 InsertDummyIslandForFetch(graph.GetFetch()); in runOnFunction()
|
| D | shape_inference.cc | 394 ValuePort(graph.GetFetch().fetches()[port[0]]), values); in ComputeOutputComponent()
932 graph_op.GetFetch(), graph_op.GetFetch().fetches(), op->getResults()); in InferShapeForNonTFDialectOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | tf_executor_to_functional.cc | 64 for (auto result : llvm::zip(graph.results(), graph.GetFetch().fetches())) in LiftIslandOpInnerOpsFromGraph()
|
| D | export_graphdef.cc | 499 for (const auto& it : llvm::enumerate(graph_op.GetFetch().getOperands())) { in Convert()
608 exporter.GetControlRetNodes(graph_op.GetFetch(), control_ret_nodes)); in Convert()
|
| D | import_model.cc | 1458 auto fetch = graph.GetFetch(); in Convert()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_executor.cc | 161 FetchOp GraphOp::GetFetch() { return llvm::cast<FetchOp>(GetBody().back()); } in GetFetch() function in mlir::tf_executor::GraphOp
967 rewriter.replaceOp(op, op.GetFetch().fetches()); in matchAndRewrite()
985 FetchOp fetch_op = op.GetFetch(); in matchAndRewrite()
|
| D | tf_executor_ops.td | 125 FetchOp GetFetch();
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/ |
| D | resource_alias_analysis.cc | 188 value = graph.GetFetch().getOperand(res_index); in BacktrackValue()
|