Searched refs:IsSend (Results 1 – 9 of 9) sorted by relevance
186 if (IsSend(*node)) { in AddNode()229 int a_score = 2 * IsSend(*a.first) + IsRecv(*a.first); in GetCurrNode()230 int b_score = 2 * IsSend(*b.first) + IsRecv(*b.first); in GetCurrNode()251 if (IsSend(*node)) { in RemoveCurrNode()350 if (IsSend(node)) { in Init()467 if ((IsSend(*node) || IsRecv(*node)) && node->attr().count(kAttrInputSrc)) { in MaybeUpdateInputOutput()611 if (IsSend(*node)) { in GetCurrNode()654 if (!IsSend(*node)) { in GetNodeStateOrCreateIt()
147 bool IsSend() const { return class_ == NC_SEND || class_ == NC_HOST_SEND; } in IsSend() function679 inline bool IsSend(const Node* node) { return node->IsSend(); } in IsSend() function685 inline bool IsTransferNode(const Node* n) { return IsSend(n) || IsRecv(n); } in IsTransferNode()
541 cnode->set_is_final(n->IsSend()); in AddToCostGraphDef()
125 if (n->IsSend() || n->IsRecv() || n->IsControlFlow()) { in Run()
107 bool IsSend(const NodeDef& node);
267 bool IsSend(const NodeDef& node) { in IsSend() function
119 TF_RET_CHECK(!n->IsRecv() && !n->IsSend() && !n->IsSwitch()) in Compile()
229 if (n->IsControlFlow() || n->IsSend() || n->IsRecv()) { in IsConstantFoldable()
100 if (IsSend(node)) { in SetTimelineLabel()749 if (IsSend(dst)) { in InferAllocAttr()