Home
last modified time | relevance | path

Searched refs:IsHostSend (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dgraph.h179 bool IsHostSend() const { return class_ == NC_HOST_SEND; } in IsHostSend() function
821 inline bool IsHostSend(const Node* node) { return node->IsHostSend(); } in IsHostSend() function
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_view.cc382 } else if ((local_dev_name.type == "CPU" || dst->IsHostSend()) && in InferAllocAttr()
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor.cc87 if (n->IsSend() || n->IsHostSend() || n->IsRecv() || n->IsHostRecv()) { in Initialize()