Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for ProcessFunctionLibraryRuntime (0.29 sec)

  1. tensorflow/compiler/jit/xla_cluster_util_test.cc

      TF_RETURN_IF_ERROR(scope.ToGraph(graph.get()));
    
      std::unique_ptr<ProcessFunctionLibraryRuntime> pflr(
          new ProcessFunctionLibraryRuntime(
              nullptr, Env::Default(), /*config=*/nullptr, TF_GRAPH_DEF_VERSION,
              flib_def, OptimizerOptions{}));
      FunctionLibraryRuntime* lib_runtime =
          pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 10.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/compilability_check_util_test.cc

      }
    
      FunctionLibraryRuntime* GetFunctionLibraryRuntime() {
        OptimizerOptions opts;
        pflr_ = std::make_unique<ProcessFunctionLibraryRuntime>(
            nullptr, Env::Default(), /*config=*/nullptr, TF_GRAPH_DEF_VERSION,
            flib_def_.get(), opts);
    
        return pflr_->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice);
      }
    
      RecursiveCompilabilityChecker::OperationFilter op_filter_;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 10 12:32:39 UTC 2022
    - 22.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/partially_decluster_pass.cc

      std::vector<bool> compile_time_const_nodes(graph->num_node_ids());
      OptimizerOptions opts;
      auto pflr = std::make_unique<ProcessFunctionLibraryRuntime>(
          nullptr, env, /*config=*/nullptr, TF_GRAPH_DEF_VERSION, flib_def, opts);
      FunctionLibraryRuntime* lib_runtime =
          pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice);
      TF_RETURN_IF_ERROR(BackwardsConstAnalysis(*graph, nullptr,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 09 11:36:41 UTC 2024
    - 15.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/get_compiler_ir.cc

     *   - `input_handles`: Contains all concrete_fn inputs tensors, including
     * captured inputs.
     */
    absl::StatusOr<std::string> GetCompilerIr(
        IrExportStage stage, ProcessFunctionLibraryRuntime* pflr,
        absl::string_view func_name, Device* dev, EagerContext* context,
        absl::Span<const ArgShapeAndDType> input_arg_shape_and_dtype,
        absl::Span<const TensorHandle* const> input_handles,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 19K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc

          FunctionLibraryDefinition *fld,
          std::vector<string> *shape_inference_graphs,
          bool *has_outside_compilation) {
        OptimizerOptions opts;
        pflr_ = std::make_unique<ProcessFunctionLibraryRuntime>(
            device_mgr_.get(), Env::Default(), /*config=*/nullptr,
            TF_GRAPH_DEF_VERSION, fld, opts,
            /*default_thread_pool=*/nullptr);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 41K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc

      }
    
      flib_def_ = std::make_unique<tensorflow::FunctionLibraryDefinition>(
          tensorflow::OpRegistry::Global(), tensorflow::FunctionDefLibrary());
      pflr_ = std::make_unique<tensorflow::ProcessFunctionLibraryRuntime>(
          device_mgr_.get(), tensorflow::Env::Default(), /*config=*/nullptr,
          version_or.value(), flib_def_.get(), tensorflow::OptimizerOptions());
      params_.function_library = pflr_->GetFLR(device_->name());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:16:07 UTC 2024
    - 18.9K bytes
    - Viewed (0)
Back to top