- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for GetCompilationDeviceType (0.38 sec)
-
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) -
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)