Searched refs:ops_to_visit (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | graph_pruning.cc | 67 llvm::SmallVectorImpl<Operation*>* ops_to_visit) { in VisitOpOperand() argument
71 ops_to_visit->push_back(def); in VisitOpOperand()
79 llvm::SmallVectorImpl<Operation*>* ops_to_visit) { in VisitOpOperands() argument
81 VisitOpOperand(graph, operand, reachable_ops, ops_to_visit); in VisitOpOperands()
90 llvm::SmallVectorImpl<Operation*>* ops_to_visit) { in VisitOp() argument
94 VisitOpOperand(graph, operand->get(), reachable_ops, ops_to_visit); in VisitOp()
98 VisitOpOperands(graph, op, reachable_ops, ops_to_visit); in VisitOp()
104 if (reachable_ops->insert(sink_op).second) ops_to_visit->push_back(sink_op); in VisitOp()
142 llvm::SmallVector<Operation*, 8> ops_to_visit; in PruneGraph() local
146 VisitOpOperands(graph, graph.GetFetch(), &reachable_ops, &ops_to_visit); in PruneGraph()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | op_selector.py | 343 ops_to_visit = [_as_operation(tensor)]
345 while ops_to_visit:
346 op = ops_to_visit.pop()
362 ops_to_visit.append(inp)
393 ops_to_visit = [_as_operation(init_tensor)]
395 while ops_to_visit:
396 op = ops_to_visit.pop()
417 ops_to_visit.append(inp)
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_communication.cc | 548 SmallVectorImpl<OpVisitorState>& ops_to_visit, in RewriteRegionIfOp() argument
586 ops_to_visit.push_back({/*region_idx=*/0, new_token, new_if}); in RewriteRegionIfOp()
596 Type block_arg_type, SmallVectorImpl<OpVisitorState>& ops_to_visit, in RewriteControlFlowOpRegion() argument
598 ops_to_visit.push_back({region_idx + 1, token, region_op}); in RewriteControlFlowOpRegion()
607 ops_to_visit.push_back({/*region_idx=*/llvm::None, block_token, in RewriteControlFlowOpRegion()
622 SmallVectorImpl<OpVisitorState>& ops_to_visit, in ProcessRegionIfOp() argument
628 RewriteRegionIfOp(builder, region_if, ops_to_visit, token); in ProcessRegionIfOp()
635 ops_to_visit, control_flow_blocks, token); in ProcessRegionIfOp()
645 SmallVectorImpl<OpVisitorState>& ops_to_visit, in RewriteRegionWhileOp() argument
681 ops_to_visit.push_back({/*region_idx=*/0, new_token, new_while}); in RewriteRegionWhileOp()
[all …]
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | lift_to_graph.py | 268 ops_to_visit = [_as_operation(t) for t in init_tensors
270 unvisited_ops = set(ops_to_visit)
272 while ops_to_visit:
273 op = ops_to_visit.pop()
286 ops_to_visit.append(inp)
292 ops_to_visit.append(next(iter(unvisited_ops)))
|