Home
last modified time | relevance | path

Searched refs:IsRecv (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dgraph.h155 bool IsRecv() const { return class_ == NC_RECV || class_ == NC_HOST_RECV; } in IsRecv() function
744 inline bool IsRecv(const Node* node) { return node->IsRecv(); } in IsRecv() function
749 inline bool IsTransferNode(const Node* n) { return IsSend(n) || IsRecv(n); } in IsTransferNode()
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc160 } else if (IsRecv(*node)) { in AddNode()
201 int a_score = 2 * IsSend(*a.first) + IsRecv(*a.first); in GetCurrNode()
202 int b_score = 2 * IsSend(*b.first) + IsRecv(*b.first); in GetCurrNode()
225 } else if (IsRecv(*node)) { in RemoveCurrNode()
358 if (IsRecv(*curr_node)) { in Init()
471 if ((IsSend(*node) || IsRecv(*node)) && node->attr().count(kAttrInputSrc)) { in MaybeUpdateInputOutput()
/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.cc87 } else if (IsRecv(node_)) { in Done()
431 if (IsSend(node) || IsRecv(node)) { in CreateNodeExecStats()
Dconstant_folding.cc234 if (n->IsControlFlow() || n->IsSend() || n->IsRecv()) { in IsConstantFoldable()
Dexecutor.cc788 if (IsRecv(n)) { in InferAllocAttr()
2015 if (!IsSwitch(node) && !IsRecv(node)) { in ProcessOutputs()
Dfunction.cc1267 if (IsRecv(e->src()) || IsSwitch(e->src())) { in GetTheOnlyDataEdge()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h133 bool IsRecv(const NodeDef& node);
Dop_types.cc395 bool IsRecv(const NodeDef& node) { in IsRecv() function
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler.cc143 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.cc409 if (n->IsSend() || n->IsRecv() || n->IsControlFlow()) { in Run()
Ddeadness_analysis.cc1029 } else if (n->IsRecv() || n->IsHostRecv()) { in HandleNode()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc76 if (IsVariable(*input) || IsRecv(*input)) { in SafeToRemoveIdentity()