- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for PjRtDeviceCompilerClient (0.62 sec)
-
tensorflow/compiler/jit/pjrt_device_compiler_client.h
void WaitForProgramsToFinish() override; xla::PjRtClient* client() const override { return client_; } private: xla::PjRtClient* const client_; PjRtDeviceCompilerClient(const PjRtDeviceCompilerClient&) = delete; void operator=(const PjRtDeviceCompilerClient&) = delete; }; // Generates CompileOptions for PJRT compilation. xla::CompileOptions GetPjRtCompileOptions( const XlaCompiler::Options& options,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_device_compiler_client.cc
return std::move(executable); } absl::StatusOr<std::string> PjRtDeviceCompilerClient::SerializeExecutable( const xla::PjRtLoadedExecutable& executable) { VLOG(1) << "Serializing xla::PjRtLoadedExecutable to string."; return executable.SerializeExecutable(); } absl::StatusOr<std::string> PjRtDeviceCompilerClient::BuildSerializedExecutable( const XlaCompiler::Options& options,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/jit/device_executable_persistor_test.cc
pjrt_compiler_client_ = std::make_unique<PjRtDeviceCompilerClient>(pjrt_client); return absl::OkStatus(); } std::unique_ptr<FunctionLibraryDefinition> flib_def_; std::unique_ptr<XlaDeviceCompilerClient> xla_compiler_client_; std::unique_ptr<PjRtDeviceCompilerClient> pjrt_compiler_client_; XlaCompiler::CompilationResult compilation_result_add_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 25.9K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_gpu_test.cc
device_compiler_ = new PjRtDeviceCompiler( std::make_unique<PjRtDeviceExecutablePersistor>( PjRtDeviceExecutablePersistor::Config(), jit_device_type), std::make_unique<PjRtDeviceCompilerClient>(pjrt_client_)); profiler_ = new DeviceCompilationProfiler(); compiler_options_.device_type = jit_device_type; compiler_options_.client = nullptr; compiler_options_.flib_def = flib_def_.get();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 10K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compiler_options_util_test.cc
auto persistor = std::make_unique<PjRtDeviceExecutablePersistor>( PjRtDeviceExecutablePersistor::Config(), device_type); auto compiler_client = std::make_unique<PjRtDeviceCompilerClient>(pjrt_client); return new PjRtDeviceCompiler(std::move(persistor), std::move(compiler_client)); } std::vector<XlaShapeLayoutHelpers::ShapeDeterminationFns>
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/xla_platform_info.cc
return new PjRtDeviceCompiler( std::make_unique<PjRtDeviceExecutablePersistor>( std::move(persistor_config), compilation_device_type), std::make_unique<PjRtDeviceCompilerClient>(pjrt_client)); } absl::StatusOr<std::optional<std::set<int>>> GetAllowedGpus( FunctionLibraryRuntime* flr) { std::optional<std::set<int>> gpu_ids = std::nullopt; if (flr->config_proto()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 17:23:27 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_test.cc
device_compiler_ = new PjRtDeviceCompiler( std::make_unique<PjRtDeviceExecutablePersistor>( PjRtDeviceExecutablePersistor::Config(), jit_device_type), std::make_unique<PjRtDeviceCompilerClient>(pjrt_client_)); profiler_ = new DeviceCompilationProfiler(); compiler_options_.device_type = jit_device_type; compiler_options_.client = nullptr; compiler_options_.flib_def = flib_def_.get();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 28.8K bytes - Viewed (0)