Searched refs:IsRecv (Results 1 – 14 of 14) sorted by relevance
165 bool IsRecv() const { return class_ == NC_RECV || class_ == NC_HOST_RECV; } in IsRecv() function820 inline bool IsRecv(const Node* node) { return node->IsRecv(); } in IsRecv() function825 inline bool IsTransferNode(const Node* n) { return IsSend(n) || IsRecv(n); } in IsTransferNode()
43 bool IsRecv(const NodeDef* node) { in IsRecv() function96 } else if (IsRecv(node_)) { in Done()429 if (IsSend(node) || IsRecv(node)) { in CreateNodeExecStats()
106 } else if (IsRecv(n)) { in Initialize()195 item->is_recv_or_switch = IsRecv(n) || IsSwitch(n); in Initialize()
114 if (IsRecv(e->src()) || IsSwitch(e->src())) { in GetTheOnlyDataEdge()
343 if (IsRecv(n)) { in InferAllocAttr()
251 if (n->IsControlFlow() || n->IsSend() || n->IsRecv()) { in IsConstantFoldable()
283 } else if (IsRecv(*node)) { in AddNode()324 int a_score = 2 * IsSend(*a.first) + IsRecv(*a.first); in GetCurrNode()325 int b_score = 2 * IsSend(*b.first) + IsRecv(*b.first); in GetCurrNode()348 } else if (IsRecv(*node)) { in RemoveCurrNode()474 if (IsRecv(*curr_node)) { in Init()593 if ((IsSend(*node) || IsRecv(*node)) && node->attr().count(kAttrInputSrc)) { in MaybeUpdateInputOutput()
149 bool IsRecv(const NodeDef& node);
466 bool IsRecv(const NodeDef& node) { in IsRecv() function
149 TF_RET_CHECK(!n->IsRecv() && !n->IsSend() && !n->IsSwitch()) in Compile()
87 if (n->IsSend() || n->IsHostSend() || n->IsRecv() || n->IsHostRecv()) { in Initialize()
580 if (n->IsSend() || n->IsRecv() || n->IsControlFlow()) { in Run()
1228 } else if (n->IsRecv() || n->IsHostRecv()) { in HandleNode()
76 if (IsVariable(*input) || IsRecv(*input)) { in SafeToRemoveIdentity()