Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for GenerateCompilerOptionsForTfrtTpu (0.34 sec)

  1. 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)
  2. 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