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