Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for GenerateCompilerOptions (0.26 sec)

  1. tensorflow/compiler/jit/xla_compiler_options_util_test.cc

      XlaPlatformInfo platform_info(
          device_type, platform_id, xla_device_metadata.get(),
          /*pjrt_device_metadata=*/nullptr, custom_allocator);
    
      XlaCompiler::Options options =
          GenerateCompilerOptions(*xla_device_compiler, *device_setup_.flr(),
                                  device, nullptr, platform_info, false);
    
      EXPECT_EQ(options.device_type, compilation_device_type);
      EXPECT_NE(options.flib_def, nullptr);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Dec 29 01:41:20 UTC 2023
    - 14.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/xla_compiler_options_util.h

    #include "xla/client/local_client.h"
    #include "xla/pjrt/pjrt_client.h"
    
    namespace tensorflow {
    
    // Returns created options for the XLA compiler.
    XlaCompiler::Options GenerateCompilerOptions(
        const DeviceCompiler<xla::LocalExecutable, xla::LocalClient>&
            xla_device_compiler,
        const FunctionLibraryRuntime& function_library, DeviceBase* device,
    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.cc

              << ",alias_passthrough_params=" << options.alias_passthrough_params
              << ",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) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 6.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/xla_compile_on_demand_op.cc

          [](DeviceCompilationProfiler** profiler) {
            *profiler = new DeviceCompilationProfiler();
            return absl::OkStatus();
          }));
    
      XlaCompiler::Options options = GenerateCompilerOptions(
          **xla_device_compiler, *ctx->function_library(), ctx->device(),
          ctx->op_device_context() ? ctx->op_device_context()->stream() : nullptr,
          platform_info_, /*has_ref_vars=*/true);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 08:39:39 UTC 2024
    - 13.4K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/get_compiler_ir.cc

      XlaCompiler::Options options;
      if (platform_info.device_type() == DEVICE_TPU && stream == nullptr) {
        options = GenerateCompilerOptionsForTfrtTpu(*xla_device_compiler, *flr);
      } else {
        options = GenerateCompilerOptions(*xla_device_compiler, *flr, dev, stream,
                                          platform_info,
                                          /*has_ref_vars=*/false);
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 19K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/kernels/xla_ops.cc

      core::ScopedUnref profiler_ref(profiler);
    
      *client = static_cast<xla::LocalClient*>(xla_device_compiler->client());
    
      XlaCompiler::Options options = GenerateCompilerOptions(
          *xla_device_compiler, *ctx->function_library(), ctx->device(),
          GetStream(ctx), platform_info, has_ref_vars);
    
      XlaCompiler::CompileOptions compile_options =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 22:46:36 UTC 2024
    - 41.4K bytes
    - Viewed (0)
Back to top