Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for platform_info_ (0.14 sec)

  1. tensorflow/compiler/jit/xla_compiler_options_util.cc

      options.graph_def_version = function_library.graph_def_version();
      options.allow_cpu_custom_calls =
          (platform_info.platform_id() == se::host::kHostPlatformId);
      options.device_allocator = GetAllocator(device, stream, platform_info);
      if (platform_info.xla_device_metadata()) {
        options.shape_determination_fns =
            platform_info.xla_device_metadata()->default_shape_determination_fns();
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 6.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/xla_platform_info.h

    // Builds a DeviceCompiler that uses xla::LocalClient using `platform_info` and
    // `compilation_device_type` (in non-TPU case) and sets *xla_device_compiler to
    // point to it. Uses flags from `MarkForCompilationPassFlags` for configuring
    // the persistor used in the DeviceCompiler. The platform ID from
    // `platform_info` must not be null in CPU case.
    Status BuildXlaDeviceCompiler(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 7.2K bytes
    - Viewed (0)
Back to top