- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for HasForwardedRefInput (0.14 sec)
-
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) -
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)