Home
last modified time | relevance | path

Searched refs:IsSend (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc248 if (IsSend(*node)) { in AddNode()
291 int a_score = 2 * IsSend(*a.first) + IsRecv(*a.first); in GetCurrNode()
292 int b_score = 2 * IsSend(*b.first) + IsRecv(*b.first); in GetCurrNode()
313 if (IsSend(*node)) { in RemoveCurrNode()
427 if (IsSend(node)) { in Init()
557 if ((IsSend(*node) || IsRecv(*node)) && node->attr().count(kAttrInputSrc)) { in MaybeUpdateInputOutput()
701 if (IsSend(*node)) { in GetCurrNode()
749 if (!IsSend(*node)) { in GetNodeStateOrCreateIt()
/external/tensorflow/tensorflow/core/graph/
Dgraph.h156 bool IsSend() const { return class_ == NC_SEND || class_ == NC_HOST_SEND; } in IsSend() function
770 inline bool IsSend(const Node* node) { return node->IsSend(); } in IsSend() function
776 inline bool IsTransferNode(const Node* n) { return IsSend(n) || IsRecv(n); } in IsTransferNode()
Dcostmodel.cc541 cnode->set_is_final(n->IsSend()); in AddToCostGraphDef()
/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.cc47 bool IsSend(const NodeDef* node) { in IsSend() function
89 if (IsSend(node_)) { in Done()
439 if (IsSend(node) || IsRecv(node)) { in CreateNodeExecStats()
Dconstant_folding.cc234 if (n->IsControlFlow() || n->IsSend() || n->IsRecv()) { in IsConstantFoldable()
Dexecutor.cc863 if (IsSend(dst)) { in InferAllocAttr()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h154 bool IsSend(const NodeDef& node);
Dop_types.cc469 bool IsSend(const NodeDef& node) { in IsSend() function
686 if (IsSend(node)) { in IsFreeOfSideEffect()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler.cc146 TF_RET_CHECK(!n->IsRecv() && !n->IsSend() && !n->IsSwitch()) in Compile()
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor.cc79 if (n->IsSend() || n->IsHostSend() || n->IsRecv() || n->IsHostRecv()) { in Initialize()
/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_ops_pass.cc576 if (n->IsSend() || n->IsRecv() || n->IsControlFlow()) { in Run()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmemory_optimizer.cc1299 if (IsSend(node)) { in FindAssignNodesToRelax()
1339 (IsSend(*fanout_node) || in FindAssignNodesToRelax()