Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for ShouldRewriteSlice (0.17 sec)

  1. 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)
Back to top