- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for ShouldRewriteSlice (0.43 sec)
-
tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass.cc
return absl::OkStatus(); } // Return true if `n` is a slice we should rewrite to have a static shape // (i.e. have the output shape only depend on the "size" input). absl::StatusOr<bool> ShouldRewriteSlice(Node* n) { if (n->type_string() != "Slice") { return false; } if (!GetXlaClusterForNode(*n).has_value()) { // There is no need to change slice ops outside XLA clusters. return false;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 13.8K bytes - Viewed (0)