Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 12 of 12 for device_allocator_ (0.27 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/get_compiler_ir.cc

                                       : local_client->default_device_ordinal());
      build_options.set_result_layout(result.xla_output_shape);
      build_options.set_device_allocator(options.device_allocator.get());
      build_options.set_alias_passthrough_params(options.alias_passthrough_params);
      build_options.mutable_debug_options()->set_xla_detailed_logging(
          options.detailed_logging);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 19K bytes
    - Viewed (0)
Back to top