- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for GenerateCompilerOptionsForPjRt (0.42 sec)
-
tensorflow/compiler/jit/xla_compiler_options_util.cc
return options; } XlaCompiler::Options GenerateCompilerOptionsForPjRt( const FunctionLibraryRuntime& function_library, const DeviceBase* device_base, const XlaPlatformInfo& platform_info, const DeviceCompiler<xla::PjRtLoadedExecutable, xla::PjRtClient>* pjrt_device_compiler) { return GenerateCompilerOptionsForPjRt( function_library.GetFunctionLibraryDefinition(),
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/xla_compiler_options_util.h
xla_device_compiler, const FunctionLibraryRuntime& function_library); // Returns created options for XLA compiler when PjRt (Device API) is used for // compilation and execution. XlaCompiler::Options GenerateCompilerOptionsForPjRt( const FunctionLibraryRuntime& function_library, const DeviceBase* device_base, const XlaPlatformInfo& platform_info, const DeviceCompiler<xla::PjRtLoadedExecutable, xla::PjRtClient>*
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_test.cc
XlaPlatformInfo platform_info( compilation_device_type, platform_id, xla_device_metadata.get(), /*pjrt_device_metadata=*/nullptr, custom_allocator); XlaCompiler::Options options = GenerateCompilerOptionsForPjRt( *device_setup_.flr(), device, platform_info, /*pjrt_device_compiler=*/nullptr); EXPECT_EQ(options.device_type, compilation_device_type); EXPECT_EQ(options.device_ordinal, 0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 29 01:41:20 UTC 2023 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_compile_util.cc
core::ScopedUnref pjrt_device_compiler_ref(pjrt_device_compiler); core::ScopedUnref profiler_ref(profiler); *client = pjrt_device_compiler->client(); XlaCompiler::Options options = GenerateCompilerOptionsForPjRt( *flr, device, platform_info, pjrt_device_compiler); XlaCompiler::CompileOptions compile_options = GenerateCompileOptions(has_ref_vars, may_alias_resource_update);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 3.8K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_on_demand_op.cc
TF_RETURN_IF_ERROR(GetOrCreatePjRtDeviceCompilerAndProfiler( *ctx, platform_info_, ctx->function_library(), pjrt_device_compiler, profiler)); XlaCompiler::Options options = GenerateCompilerOptionsForPjRt(*(ctx->function_library()), ctx->device(), platform_info_, *pjrt_device_compiler); // No detailed logging for on demand op. options.detailed_logging = false;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 08:39:39 UTC 2024 - 13.4K bytes - Viewed (0)