| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | traversal.cc | 41 enum class NodeState { kNotVisited, kVisiting, kDone }; enum
61 absl::flat_hash_map<int, NodeState> node_state; in DfsTraversal()
66 NodeState& state = node_state[w.node]; in DfsTraversal()
67 if (state == NodeState::kDone) continue; in DfsTraversal()
71 state = NodeState::kDone; in DfsTraversal()
77 state = NodeState::kDone; in DfsTraversal()
85 if (state == NodeState::kVisiting) { in DfsTraversal()
93 state = NodeState::kVisiting; in DfsTraversal()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler.h | 42 struct NodeState { struct
90 NodeState() { in NodeState() function
166 const std::unordered_map<const NodeDef*, NodeState>* node_map) { in Init()
220 const std::unordered_map<const NodeDef*, NodeState>* node_map) override;
240 const std::unordered_map<const NodeDef*, NodeState>* node_map_;
290 const std::unordered_map<const NodeDef*, NodeState>* node_map) override;
308 const std::unordered_map<const NodeDef*, NodeState>* node_map_;
368 const std::unordered_map<const NodeDef*, NodeState>* GetNodeStates() const { in GetNodeStates()
385 const std::unordered_map<const NodeDef*, NodeState>& GetNodeMap() { in GetNodeMap()
404 NodeState& GetNodeStateOrCreateIt(const NodeDef* node);
[all …]
|
| D | virtual_scheduler.cc | 94 const NodeState& node_state, in UpdateDeviceAnnotationState()
166 const std::unordered_map<const NodeDef*, NodeState>* node_map) { in Init()
220 const std::unordered_map<const NodeDef*, NodeState>* node_map, in FirstReadyCmp()
272 const std::unordered_map<const NodeDef*, NodeState>* node_map) { in Init()
777 NodeState& SchedulerState::GetNodeStateOrCreateIt(const NodeDef* node) { in GetNodeStateOrCreateIt()
786 it = node_map_.emplace(node, NodeState()).first; in GetNodeStateOrCreateIt()
1210 const NodeState& nodestate = node_map_.at(node_def); in GenerateRunMetadata()
|
| D | virtual_scheduler_test.cc | 60 node_states_[node] = NodeState(); in NodeSetUp()
66 std::unordered_map<const NodeDef*, NodeState> node_states_;
2654 const NodeState* bn_node = nullptr; in TEST_F()
2655 const NodeState* x_node = nullptr; in TEST_F()
2658 const NodeState& node_state = nodedef_node_state.second; in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | eager_executor.cc | 132 item->state = NodeState::kPENDING; in AddOrExecute()
218 DCHECK(item->state != NodeState::kDONE); in NodeDone()
219 item->state = NodeState::kDONE; in NodeDone()
390 item->state = NodeState::kSCHEDULED; in RunItem()
|
| D | eager_executor.h | 179 enum class NodeState { enum
190 NodeState state;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | pending_counts.h | 53 enum NodeState { enum
121 NodeState node_state(Handle h) { in node_state()
349 NodeState NodeStateForStruct(const T& c) const { in NodeStateForStruct()
|
| D | propagator_state.h | 161 PendingCounts::NodeState node_state(PendingCounts::Handle h) { in node_state()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager_test.cc | 297 struct NodeState { struct in tensorflow::NcclManagerTest
304 std::vector<NodeState> node_states(num_nodes); in RunMultiNodeAllReduceTest()
308 void RunMultiNodeAllReduceTest(std::vector<NodeState>& node_states, in RunMultiNodeAllReduceTest()
369 std::vector<NodeState> node_states(num_nodes); in RunMultiNodeBroadcastTest()
866 using NodeState = typename TestFixture::NodeState; in TYPED_TEST() typedef
869 std::vector<NodeState> nodes(num_nodes); in TYPED_TEST()
|
| /external/llvm/include/llvm/Object/ |
| D | MachO.h | 81 struct NodeState { struct
82 NodeState(const uint8_t *Ptr);
97 SmallVector<NodeState, 16> Stack; argument
|
| /external/llvm-project/llvm/include/llvm/Object/ |
| D | MachO.h | 97 struct NodeState { struct
98 NodeState(const uint8_t *Ptr);
111 using NodeList = SmallVector<NodeState, 16>; argument
|
| /external/swiftshader/third_party/llvm-10.0/llvm/include/llvm/Object/ |
| D | MachO.h | 97 struct NodeState { struct
98 NodeState(const uint8_t *Ptr);
111 using NodeList = SmallVector<NodeState, 16>; argument
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | auto_mixed_precision.cc | 745 enum class NodeState { kNotVisited, kVisiting, kDone }; enum
764 absl::flat_hash_map<int, NodeState> node_state; in DfsTypeTraversal()
769 NodeState& state = node_state[w.node]; in DfsTypeTraversal()
770 if (state == NodeState::kDone) continue; in DfsTypeTraversal()
774 state = NodeState::kDone; in DfsTypeTraversal()
780 state = NodeState::kDone; in DfsTypeTraversal()
788 if (state == NodeState::kVisiting) { in DfsTypeTraversal()
795 state = NodeState::kVisiting; in DfsTypeTraversal()
|
| /external/llvm/lib/Object/ |
| D | MachOObjectFile.cpp | 1405 ExportEntry::NodeState::NodeState(const uint8_t *Ptr) in NodeState() function in ExportEntry::NodeState
1412 NodeState State(Ptr); in pushNode()
1437 NodeState &Top = Stack.back(); in pushDownUntilBottom()
1478 NodeState &Top = Stack.back(); in moveNext()
|
| /external/swiftshader/third_party/llvm-10.0/llvm/lib/Object/ |
| D | MachOObjectFile.cpp | 2854 ExportEntry::NodeState::NodeState(const uint8_t *Ptr) in NodeState() function in ExportEntry::NodeState
2860 NodeState State(Ptr); in pushNode()
2996 NodeState &Top = Stack.back(); in pushDownUntilBottom()
3019 for (const NodeState &node : nodes()) { in pushDownUntilBottom()
3070 NodeState &Top = Stack.back(); in moveNext()
|
| /external/llvm-project/llvm/lib/Object/ |
| D | MachOObjectFile.cpp | 2885 ExportEntry::NodeState::NodeState(const uint8_t *Ptr) in NodeState() function in ExportEntry::NodeState
2891 NodeState State(Ptr); in pushNode()
3027 NodeState &Top = Stack.back(); in pushDownUntilBottom()
3050 for (const NodeState &node : nodes()) { in pushDownUntilBottom()
3101 NodeState &Top = Stack.back(); in moveNext()
|