Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for GetOnlyNodeIn (0.28 sec)

  1. tensorflow/compiler/jit/mark_for_compilation_pass.cc

        }
      }
    
      return changed;
    }
    
    Node* MarkForCompilationPassImpl::GetOnlyNodeIn(const Cluster& cluster) {
      return cluster.cluster_size() == 1
                 ? graph_->FindNodeId(cluster.GetIdOfOnlyNode())
                 : nullptr;
    }
    
    bool MarkForCompilationPassImpl::IsSinkLike(const Cluster& cluster) {
      if (Node* n = GetOnlyNodeIn(cluster)) {
        return n->type_string() == "NoOp" && n->out_edges().size() == 1 &&
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
Back to top