Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for CreateXlaDeviceCompiler (0.24 sec)

  1. tensorflow/compiler/jit/device_compiler_test.cc

      // well.
      auto platform = se::PlatformManager::PlatformWithName("cuda").value();
      return xla::ClientLibrary::GetOrCreateLocalClient(platform).value();
    }
    
    XlaDeviceCompiler* CreateXlaDeviceCompiler(bool enable_persistence = false) {
      auto xla_compiler_client =
          std::make_unique<XlaDeviceCompilerClient>(GetLocalClient());
      auto xla_persistor = std::make_unique<XlaDeviceExecutablePersistor>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 19.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/xla_compiler_options_util_test.cc

        DeviceCompiler<xla::PjRtLoadedExecutable, xla::PjRtClient>;
    using PjRtDeviceExecutablePersistor =
        DeviceExecutablePersistor<xla::PjRtLoadedExecutable, xla::PjRtClient>;
    
    XlaDeviceCompiler* CreateXlaDeviceCompiler(DeviceType device_type,
                                               xla::LocalClient* local_client) {
      auto persistor = std::make_unique<XlaDeviceExecutablePersistor>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Dec 29 01:41:20 UTC 2023
    - 14.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/xla_platform_info.cc

        DeviceExecutablePersistor<xla::LocalExecutable, xla::LocalClient>;
    using PjRtDeviceExecutablePersistor =
        DeviceExecutablePersistor<xla::PjRtLoadedExecutable, xla::PjRtClient>;
    
    XlaDeviceCompiler* CreateXlaDeviceCompiler(
        const XlaDeviceExecutablePersistor::Config& persistor_config,
        DeviceType compilation_device_type, xla::LocalClient* local_client) {
      return new XlaDeviceCompiler(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 17:23:27 UTC 2024
    - 17.4K bytes
    - Viewed (0)
Back to top