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

Function HasForwardedRefInput

tensorflow/compiler/jit/xla_cluster_util.cc:88–104  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

86} // namespace
87
88bool HasForwardedRefInput(const Node& node) {
89 if (AlwaysForwardsRefInput(node)) {
90 for (const Edge* incoming_edge : node.in_edges()) {
91 if (incoming_edge->IsControlEdge()) {
92 continue;
93 }
94
95 Node* incoming_node = incoming_edge->src();
96 if (IsRefType(incoming_node->output_type(incoming_edge->src_output()))) {
97 VLOG(2) << "Node " << node.def().ShortDebugString() << " has ref input "
98 << incoming_node->name() << " " << incoming_node->type_string();
99 return true;
100 }
101 }
102 }
103 return false;
104}
105
106xla::StatusOr<bool> CreateCycleDetectionGraph(const Graph* graph,
107 GraphCycles* cycles) {

Callers 1

HasXLAKernelMethod · 0.85

Calls 8

IsRefTypeFunction · 0.85
AlwaysForwardsRefInputFunction · 0.70
nameMethod · 0.65
IsControlEdgeMethod · 0.45
srcMethod · 0.45
output_typeMethod · 0.45
src_outputMethod · 0.45
ShortDebugStringMethod · 0.45

Tested by

no test coverage detected