Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for FunctionLibraryRuntime (0.33 sec)

  1. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

      std::unique_ptr<Graph> graph_out;
      FunctionLibraryDefinition* const library = options.flib_def;
    
      // Constant folding below might need to run part of the function to compute
      // constants. Create an FunctionLibraryRuntime with a single CPU device
      // that can run the part of the function.
      // NOTE: If this turns out to be slow, we can cache the FLRs keyed by
      // `options`.
      SessionOptions session_options;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 51K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/mark_for_compilation_pass.cc

      std::unique_ptr<ProcessFunctionLibraryRuntime> pflr(
          new ProcessFunctionLibraryRuntime(nullptr, env_, /*config=*/nullptr,
                                            TF_GRAPH_DEF_VERSION, flib_def_, opts));
      FunctionLibraryRuntime* lib_runtime =
          pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice);
      std::vector<bool> compile_time_const_nodes(graph_->num_node_ids(), false);
      TF_RETURN_IF_ERROR(BackwardsConstAnalysis(
    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