Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for GetCompilationDeviceType (0.38 sec)

  1. tensorflow/compiler/jit/xla_platform_info.cc

              platform_device_id_str,
              "'. visible_device_list = ", visible_device_list);
        }
        gpu_ids.insert(platform_device_id);
      }
      return {{gpu_ids}};
    }
    
    absl::StatusOr<DeviceType> GetCompilationDeviceType(
        const DeviceType& platform_device_type) {
      DeviceType compilation_device_type = platform_device_type;
      const XlaOpRegistry::DeviceRegistration* registration = nullptr;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 17:23:27 UTC 2024
    - 17.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/kernels/xla_ops.cc

      ResourceMgr* rm = ctx->resource_manager();
      if (!rm) {
        return absl::InternalError("No resource manager.");
      }
    
      TF_ASSIGN_OR_RETURN(DeviceType compilation_device_type,
                          GetCompilationDeviceType(platform_info.device_type()));
    
      XlaDeviceCompiler* xla_device_compiler;
      TF_RETURN_IF_ERROR(rm->LookupOrCreate<XlaDeviceCompiler>(
          rm->default_container(), "xla_device_compiler", &xla_device_compiler,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 22:46:36 UTC 2024
    - 41.4K bytes
    - Viewed (0)
Back to top