Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for GenerateCompilerOptionsForTfrtTpu (0.43 sec)

  1. tensorflow/compiler/jit/xla_compiler_options_util.h

        se::Stream* stream, const XlaPlatformInfo& platform_info,
        bool has_ref_vars);
    
    // Returns created options for XLA compiler when TFRT-TPU is used.
    XlaCompiler::Options GenerateCompilerOptionsForTfrtTpu(
        const DeviceCompiler<xla::LocalExecutable, xla::LocalClient>&
            xla_device_compiler,
        const FunctionLibraryRuntime& function_library);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Dec 29 01:41:20 UTC 2023
    - 2.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/xla_compiler_options_util.cc

      options.alias_passthrough_params =
          !has_ref_vars && !platform_info.is_on_xla_device();
    
      LogOptions(options);
      return options;
    }
    
    XlaCompiler::Options GenerateCompilerOptionsForTfrtTpu(
        const XlaDeviceCompiler& xla_device_compiler,
        const FunctionLibraryRuntime& function_library) {
      XlaCompiler::Options options;
      // TODO(b/238830423): consider device_ordinal and shape_determination_fns.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 6.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/get_compiler_ir.cc

          }));
      core::ScopedUnref xla_device_compiler_ref(xla_device_compiler);
    
      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,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 19K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/xla_compiler_options_util_test.cc

      auto xla_device_compiler =
          CreateXlaDeviceCompiler(compilation_device_type, client);
      core::ScopedUnref xla_device_compiler_ref(xla_device_compiler);
    
      XlaCompiler::Options options = GenerateCompilerOptionsForTfrtTpu(
          *xla_device_compiler, *device_setup_.flr());
    
      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)
Back to top