Searched refs:IsRecv (Results 1 – 13 of 13) sorted by relevance
155 bool IsRecv() const { return class_ == NC_RECV || class_ == NC_HOST_RECV; } in IsRecv() function744 inline bool IsRecv(const Node* node) { return node->IsRecv(); } in IsRecv() function749 inline bool IsTransferNode(const Node* n) { return IsSend(n) || IsRecv(n); } in IsTransferNode()
160 } else if (IsRecv(*node)) { in AddNode()201 int a_score = 2 * IsSend(*a.first) + IsRecv(*a.first); in GetCurrNode()202 int b_score = 2 * IsSend(*b.first) + IsRecv(*b.first); in GetCurrNode()225 } else if (IsRecv(*node)) { in RemoveCurrNode()358 if (IsRecv(*curr_node)) { in Init()471 if ((IsSend(*node) || IsRecv(*node)) && node->attr().count(kAttrInputSrc)) { in MaybeUpdateInputOutput()
87 } else if (IsRecv(node_)) { in Done()431 if (IsSend(node) || IsRecv(node)) { in CreateNodeExecStats()
234 if (n->IsControlFlow() || n->IsSend() || n->IsRecv()) { in IsConstantFoldable()
788 if (IsRecv(n)) { in InferAllocAttr()2015 if (!IsSwitch(node) && !IsRecv(node)) { in ProcessOutputs()
1267 if (IsRecv(e->src()) || IsSwitch(e->src())) { in GetTheOnlyDataEdge()
133 bool IsRecv(const NodeDef& node);
395 bool IsRecv(const NodeDef& node) { in IsRecv() function
143 TF_RET_CHECK(!n->IsRecv() && !n->IsSend() && !n->IsSwitch()) in Compile()
79 if (n->IsSend() || n->IsHostSend() || n->IsRecv() || n->IsHostRecv()) { in Initialize()
409 if (n->IsSend() || n->IsRecv() || n->IsControlFlow()) { in Run()
1029 } else if (n->IsRecv() || n->IsHostRecv()) { in HandleNode()
76 if (IsVariable(*input) || IsRecv(*input)) { in SafeToRemoveIdentity()