Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for CreateXlaDeviceCompiler (0.17 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_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