- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for DeviceExecutablePersistor (0.56 sec)
-
tensorflow/compiler/jit/device_executable_persistor.h
const bool persistent_cache_directory_read_only_; DeviceExecutablePersistor(const DeviceExecutablePersistor&) = delete; void operator=(const DeviceExecutablePersistor&) = delete; }; template <typename ExecutableType, typename ClientType> DeviceExecutablePersistor<ExecutableType, ClientType>:: DeviceExecutablePersistor(const Config& config, const DeviceType& device_type)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/jit/device_compiler.h
// calling `ClientType` (eg. xla::LocalClient). // // Caches the compiled XlaCompilationResult and Executable using a // DeviceCompilationCache. Compilation is done only when there's a cache miss. // // Uses the DeviceExecutablePersistor class for persistence and tries to load a // serialized executable from disk upon a request for compilation. If the // appropriate executable isn't found on disk, compiles the given Tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 22.1K bytes - Viewed (0) -
tensorflow/compiler/jit/device_compiler_disable_test.cc
DeviceType device_type = DeviceType(DEVICE_CPU_XLA_JIT); const XlaCompiler::CompilationResult* compilation_result; xla::LocalExecutable* executable; using XlaDeviceExecutablePersistor = DeviceExecutablePersistor<xla::LocalExecutable, xla::LocalClient>; auto persistor = std::make_unique<XlaDeviceExecutablePersistor>( XlaDeviceExecutablePersistor::Config(), device_type);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/jit/device_compiler_test.cc
args[1].type = DT_INT32; args[1].shape = TensorShape({2}); return args; } class MockXlaDeviceExecutablePersistor : public DeviceExecutablePersistor<xla::LocalExecutable, xla::LocalClient> { public: MockXlaDeviceExecutablePersistor() : DeviceExecutablePersistor<xla::LocalExecutable, xla::LocalClient>( Config{testing::TmpDir(), false, "xla"}, DeviceType(DEVICE_CPU_XLA_JIT)) {}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 19.8K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compiler_options_util_test.cc
DeviceCompiler<xla::LocalExecutable, xla::LocalClient>; using XlaDeviceExecutablePersistor = DeviceExecutablePersistor<xla::LocalExecutable, xla::LocalClient>; using PjRtDeviceCompiler = DeviceCompiler<xla::PjRtLoadedExecutable, xla::PjRtClient>; using PjRtDeviceExecutablePersistor = DeviceExecutablePersistor<xla::PjRtLoadedExecutable, xla::PjRtClient>; XlaDeviceCompiler* CreateXlaDeviceCompiler(DeviceType device_type,
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
using PjRtDeviceCompiler = DeviceCompiler<xla::PjRtLoadedExecutable, xla::PjRtClient>; using XlaDeviceExecutablePersistor = DeviceExecutablePersistor<xla::LocalExecutable, xla::LocalClient>; using PjRtDeviceExecutablePersistor = DeviceExecutablePersistor<xla::PjRtLoadedExecutable, xla::PjRtClient>; XlaDeviceCompiler* CreateXlaDeviceCompiler( const XlaDeviceExecutablePersistor::Config& persistor_config,
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/device_executable_persistor_test.cc
namespace tensorflow { namespace { using ::testing::_; using ::testing::ByMove; using ::testing::Return; using XlaDeviceExecutablePersistor = DeviceExecutablePersistor<xla::LocalExecutable, xla::LocalClient>; using PjRtDeviceExecutablePersistor = DeviceExecutablePersistor<xla::PjRtLoadedExecutable, xla::PjRtClient>; class DeviceExecutionPersistorTest : public ::testing::Test { protected: void SetUp() override {
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
#include "tsl/platform/statusor.h" namespace tensorflow { namespace { using PjRtDeviceCompiler = DeviceCompiler<xla::PjRtLoadedExecutable, xla::PjRtClient>; using PjRtDeviceExecutablePersistor = DeviceExecutablePersistor<xla::PjRtLoadedExecutable, xla::PjRtClient>; absl::flat_hash_map<int, const Tensor*> GetVariableSnapshots( const std::vector<VariableInfo>& variables) { absl::flat_hash_map<int, const Tensor*> variable_snapshots;
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_launch_util_test.cc
#include "tsl/platform/statusor.h" namespace tensorflow { namespace { using PjRtDeviceCompiler = DeviceCompiler<xla::PjRtLoadedExecutable, xla::PjRtClient>; using PjRtDeviceExecutablePersistor = DeviceExecutablePersistor<xla::PjRtLoadedExecutable, xla::PjRtClient>; absl::flat_hash_map<int, const Tensor*> GetVariableSnapshots( const std::vector<VariableInfo>& variables) { absl::flat_hash_map<int, const Tensor*> variable_snapshots;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 28.8K bytes - Viewed (0)