Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for GetGlobalJitLevelForGraph (0.29 sec)

  1. tensorflow/compiler/jit/xla_cluster_util.h

    bool HasResourceInputOrOutput(const Node& node);
    
    // Determines the global jit level based on GraphOptimizationPassOptions,
    // --tf_xla_auto_jit and whether the graph is a single GPU graph.
    OptimizerOptions::GlobalJitLevel GetGlobalJitLevelForGraph(
        const GraphOptimizationPassOptions& options);
    
    // Returns true if `g` is a single-GPU graph.  A single-GPU graph uses exactly
    // one GPU (and any number of CPUs).
    bool IsSingleGpuGraph(const Graph& g);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 4.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/xla_cluster_util.cc

        VLOG(4) << "GetGlobalJitLevelForGraph returning "
                << xla_global_jit_level.single_gpu;
        return xla_global_jit_level.single_gpu;
      }
      OptimizerOptions::GlobalJitLevel result =
          IsSingleGpuGraph(**options.graph) ? xla_global_jit_level.single_gpu
                                            : xla_global_jit_level.general;
      VLOG(4) << "GetGlobalJitLevelForGraph returning " << result;
      return result;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 08:39:39 UTC 2024
    - 21.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/clone_constants_for_better_clustering.cc

      }
      return absl::OkStatus();
    }
    
    Status CloneConstantsForBetterClusteringPass::Run(
        const GraphOptimizationPassOptions& options) {
      if (GetGlobalJitLevelForGraph(options) == OptimizerOptions::OFF) {
        return absl::OkStatus();
      }
    
      Graph* g = options.graph->get();
    
      if (VLOG_IS_ON(1)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/cluster_scoping_pass.cc

    }
    }  // namespace
    
    Status ClusterScopingPass::Run(const GraphOptimizationPassOptions& options) {
      Graph* graph = options.graph->get();
    
      return ClusterScopingPassImpl{graph, GetGlobalJitLevelForGraph(options)}
          .Run();
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 5.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/mark_for_compilation_pass.cc

          debug_options,
          graph,
          flib_def,
          options.session_options != nullptr ? options.session_options->env
                                             : Env::Default(),
          GetGlobalJitLevelForGraph(options),
          options.session_options->config.graph_options()
              .optimizer_options()
              .cpu_global_jit(),
          /*cluster_name_prefix=*/options.session_options != nullptr
    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