Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for FunctionLibraryRuntime (0.32 sec)

  1. tensorflow/compiler/jit/xla_compiler_options_util.cc

              << ",detailed_logging=" << options.detailed_logging << "]";
    }
    }  // namespace
    
    XlaCompiler::Options GenerateCompilerOptions(
        const XlaDeviceCompiler& xla_device_compiler,
        const FunctionLibraryRuntime& function_library, DeviceBase* device,
        se::Stream* stream, const XlaPlatformInfo& platform_info,
        bool has_ref_vars) {
      XlaCompiler::Options options;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 6.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/xla_platform_info.h

    // the persistor used in the DeviceCompiler. The platform ID from
    // `platform_info` must not be null in CPU case.
    Status BuildXlaDeviceCompiler(
        DeviceBase* dev, FunctionLibraryRuntime* flr,
        const XlaPlatformInfo& platform_info, DeviceType compilation_device_type,
        DeviceCompiler<xla::LocalExecutable, xla::LocalClient>**
            xla_device_compiler);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 7.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/force_xla_constants_on_host_pass.cc

      OptimizerOptions opts;
      auto pflr = std::make_unique<ProcessFunctionLibraryRuntime>(
          nullptr, options.session_options->env, /*config=*/nullptr,
          TF_GRAPH_DEF_VERSION, options.flib_def, opts);
      FunctionLibraryRuntime* flr =
          pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice);
    
      for (Node* node : graph->nodes()) {
        if (CanCreateXlaKernel(node->def())) {
          const FunctionBody* fbody = nullptr;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 2.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/xla_cluster_util.h

    // all of the nodes that have ref variables as input or output.
    absl::StatusOr<absl::flat_hash_set<Node*>> GetNodesRelatedToRefVariables(
        const Graph& graph, FunctionLibraryRuntime* lib_runtime);
    
    // Deterministically serialized the graph to a byte string.
    absl::StatusOr<std::string> SerializeGraphDeterministic(const Graph& graph);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 4.6K bytes
    - Viewed (0)
Back to top