Searched refs:IsHostRecv (Results 1 – 2 of 2) sorted by relevance
163 bool IsHostRecv() const { return class_ == NC_HOST_RECV; } in IsHostRecv() function682 inline bool IsHostRecv(const Node* node) { return node->IsHostRecv(); } in IsHostRecv() function
738 } else if ((local_dev_name.type == "CPU" || n->IsHostRecv()) && in InferAllocAttr()