HomeSort by relevance Sort by last modified time
    Searched refs:IsRecv (Results 1 - 10 of 10) sorted by null

  /external/tensorflow/tensorflow/core/grappler/costs/
virtual_scheduler.cc 188 } else if (IsRecv(*node)) {
229 int a_score = 2 * IsSend(*a.first) + IsRecv(*a.first);
230 int b_score = 2 * IsSend(*b.first) + IsRecv(*b.first);
253 } else if (IsRecv(*node)) {
367 if (IsRecv(*curr_node)) {
431 if (!IsRecv(*curr_node) && (given_as_feed || has_no_inputs)) {
467 if ((IsSend(*node) || IsRecv(*node)) && node->attr().count(kAttrInputSrc)) {
    [all...]
  /external/tensorflow/tensorflow/compiler/jit/
build_xla_launch_ops_pass.cc 125 if (n->IsSend() || n->IsRecv() || n->IsControlFlow()) {
  /external/tensorflow/tensorflow/core/graph/
graph.h 148 bool IsRecv() const { return class_ == NC_RECV || class_ == NC_HOST_RECV; }
680 inline bool IsRecv(const Node* node) { return node->IsRecv(); }
685 inline bool IsTransferNode(const Node* n) { return IsSend(n) || IsRecv(n); }
  /external/tensorflow/tensorflow/compiler/tf2xla/
graph_compiler.cc 119 TF_RET_CHECK(!n->IsRecv() && !n->IsSend() && !n->IsSwitch())
  /external/tensorflow/tensorflow/core/grappler/
op_types.h 99 bool IsRecv(const NodeDef& node);
op_types.cc 238 bool IsRecv(const NodeDef& node) {
  /external/tensorflow/tensorflow/core/common_runtime/
constant_folding.cc 229 if (n->IsControlFlow() || n->IsSend() || n->IsRecv()) {
executor.cc 108 } else if (IsRecv(node)) {
724 if (IsRecv(n)) {
    [all...]
function.cc     [all...]
  /external/tensorflow/tensorflow/core/grappler/optimizers/
dependency_optimizer.cc 84 if (IsVariable(*input) || IsRecv(*input)) {

Completed in 354 milliseconds