- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for GenerateCompilerOptionsForTfrtTpu (0.34 sec)
-
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) -
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)