Searched refs:IsHostRecv (Results 1 – 4 of 4) sorted by relevance
172 bool IsHostRecv() const { return class_ == NC_HOST_RECV; } in IsHostRecv() function773 inline bool IsHostRecv(const Node* node) { return node->IsHostRecv(); } in IsHostRecv() function
79 if (n->IsSend() || n->IsHostSend() || n->IsRecv() || n->IsHostRecv()) { in Initialize()
1228 } else if (n->IsRecv() || n->IsHostRecv()) { in HandleNode()
852 } else if ((local_dev_name.type == "CPU" || n->IsHostRecv()) && in InferAllocAttr()