MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / IsSplitedIgnored

Method IsSplitedIgnored

tensorflow/core/graph/star_server_graph_partition.cc:131–156  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

129}
130
131bool GraphPartitionerBase::IsSplitedIgnored(
132 const std::string &cur_loc,
133 const Node* node)
134{
135 if (!node->IsOp()) {
136 return true;
137 }
138
139 if (is_main_loc_func_(cur_loc)) {
140 return false;
141 }
142
143 if (node->IsKvVarHandle ()) {
144 return true;
145 }
146
147 if (node->IsVariable()) {
148 return true;
149 }
150
151 if (node->IsIdentity() && GetVarOfIdentity(opts_, node) != nullptr) {
152 return true;
153 }
154
155 return false;
156}
157
158void GraphPartitionerBase::TryAddEdgeNode(
159 const std::string &cur_loc, const Node *node, deque<const Node*> &nodes,

Callers

nothing calls this directly

Calls 5

GetVarOfIdentityFunction · 0.85
IsOpMethod · 0.80
IsKvVarHandleMethod · 0.80
IsVariableMethod · 0.80
IsIdentityMethod · 0.80

Tested by

no test coverage detected