- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for PjRtDeviceCompilerClient (0.29 sec)
-
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)