Searched refs:kVisiting (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | traversal.cc | 41 enum class NodeState { kNotVisited, kVisiting, kDone }; enumerator
85 if (state == NodeState::kVisiting) { in DfsTraversal()
93 state = NodeState::kVisiting; in DfsTraversal()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_group_util.cc | 273 if ((*visit_state)[hlo] == VisitState::kVisiting) { in VisitTopologicalOrder()
278 TF_RET_CHECK((*visit_state)[instruction] == VisitState::kVisiting); in VisitTopologicalOrder()
289 (*visit_state)[instruction] = VisitState::kVisiting; in VisitTopologicalOrder()
299 if ((*visit_state)[predecessor] == VisitState::kVisiting) { in VisitTopologicalOrder()
|
| D | dfs_hlo_visitor.cc | 53 visit_state_[instruction.unique_id()] = VisitState::kVisiting; in SetVisiting()
|
| D | hlo_module_group_util.h | 64 kVisiting, enumerator
|
| D | dfs_hlo_visitor.h | 304 kVisiting = 1, enumerator
352 return GetVisitState(instruction) == kVisiting; in IsVisiting()
|
| D | hlo_computation.cc | 454 auto result = visited.insert({¤t, kVisiting}); in ComputeInstructionPostOrder()
1247 CHECK_EQ(it->second, kVisiting); in CloneWithReplacements()
1253 visited.insert({cur, kVisiting}); in CloneWithReplacements()
|
| D | hlo_computation.h | 696 enum VisitState { kVisiting, kVisited }; enumerator
|
| D | hlo_instruction.cc | 3727 case Visitor::kVisiting: in PushDFSChild()
3774 if (visit_state == Visitor::kVisiting) { in PostOrderDFS()
3785 visitor->SetVisitState(current_id, Visitor::kVisiting); in PostOrderDFS()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | value_inference.cc | 1493 kVisiting, in PostOrderDFSVisit() enumerator
1530 if (item.state == kVisiting) { in PostOrderDFSVisit()
1556 item.state = kVisiting; in PostOrderDFSVisit()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | auto_mixed_precision.cc | 811 enum class NodeState { kNotVisited, kVisiting, kDone }; enumerator
854 if (state == NodeState::kVisiting) { in DfsTypeTraversal()
861 state = NodeState::kVisiting; in DfsTypeTraversal()
|