Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for HasForwardedRefInput (0.18 sec)

  1. tensorflow/compiler/jit/xla_cluster_util.cc

        absl::StrAppend(&description, ascii_art, node_name(node_id), "\n");
      }
      return description;
    }
    
    bool AlwaysForwardsRefInput(const Node& node) { return node.IsIdentity(); }
    
    }  // namespace
    
    bool HasForwardedRefInput(const Node& node) {
      if (AlwaysForwardsRefInput(node)) {
        for (const Edge* incoming_edge : node.in_edges()) {
          if (incoming_edge->IsControlEdge()) {
            continue;
          }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 08:39:39 UTC 2024
    - 21.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/compilability_check_util.cc

      }
    
      // XLA does not offer guaranteed aliasing between the input and output of the
      // XLA cluster so it can't implement the forward-tensor-ref semantic.  Leave
      // such nodes out of XLA clusters.
      if (HasForwardedRefInput(node)) {
        VLOG(2) << "Rejecting " << node.name() << ": Identity with unsafe cast.";
        *uncompilable_reason = "Identity with unsafe cast.";
        return false;
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 30.3K bytes
    - Viewed (0)
Back to top