Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for GenerateCompilerOptionsForTfrtTpu (0.32 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)
Back to top