Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for GenerateCompilerOptionsForPjRt (0.55 sec)

  1. tensorflow/compiler/jit/xla_compiler_options_util.cc

      return options;
    }
    
    XlaCompiler::Options GenerateCompilerOptionsForPjRt(
        const FunctionLibraryRuntime& function_library,
        const DeviceBase* device_base, const XlaPlatformInfo& platform_info,
        const DeviceCompiler<xla::PjRtLoadedExecutable, xla::PjRtClient>*
            pjrt_device_compiler) {
      return GenerateCompilerOptionsForPjRt(
          function_library.GetFunctionLibraryDefinition(),
    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_compiler_options_util.h

            xla_device_compiler,
        const FunctionLibraryRuntime& function_library);
    
    // Returns created options for XLA compiler when PjRt (Device API) is used for
    // compilation and execution.
    XlaCompiler::Options GenerateCompilerOptionsForPjRt(
        const FunctionLibraryRuntime& function_library,
        const DeviceBase* device_base, const XlaPlatformInfo& platform_info,
        const DeviceCompiler<xla::PjRtLoadedExecutable, xla::PjRtClient>*
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Dec 29 01:41:20 UTC 2023
    - 2.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/xla_compiler_options_util_test.cc

      XlaPlatformInfo platform_info(
          compilation_device_type, platform_id, xla_device_metadata.get(),
          /*pjrt_device_metadata=*/nullptr, custom_allocator);
    
      XlaCompiler::Options options = GenerateCompilerOptionsForPjRt(
          *device_setup_.flr(), device, platform_info,
          /*pjrt_device_compiler=*/nullptr);
    
      EXPECT_EQ(options.device_type, compilation_device_type);
      EXPECT_EQ(options.device_ordinal, 0);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Dec 29 01:41:20 UTC 2023
    - 14.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/pjrt_compile_util.cc

      core::ScopedUnref pjrt_device_compiler_ref(pjrt_device_compiler);
      core::ScopedUnref profiler_ref(profiler);
    
      *client = pjrt_device_compiler->client();
    
      XlaCompiler::Options options = GenerateCompilerOptionsForPjRt(
          *flr, device, platform_info, pjrt_device_compiler);
    
      XlaCompiler::CompileOptions compile_options =
          GenerateCompileOptions(has_ref_vars, may_alias_resource_update);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 3.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/xla_compile_on_demand_op.cc

      TF_RETURN_IF_ERROR(GetOrCreatePjRtDeviceCompilerAndProfiler(
          *ctx, platform_info_, ctx->function_library(), pjrt_device_compiler,
          profiler));
    
      XlaCompiler::Options options =
          GenerateCompilerOptionsForPjRt(*(ctx->function_library()), ctx->device(),
                                         platform_info_, *pjrt_device_compiler);
      // No detailed logging for on demand op.
      options.detailed_logging = false;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 08:39:39 UTC 2024
    - 13.4K bytes
    - Viewed (0)
Back to top