- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for GetCompilationDeviceType (0.35 sec)
-
tensorflow/compiler/jit/xla_platform_info_test.cc
TF_CHECK_OK(XlaDevice::GetMetadataFromDevice(device, &metadata)); XlaPlatformInfo platform_info = XlaPlatformInfoFromDevice(device); TF_ASSERT_OK_AND_ASSIGN( DeviceType compilation_device_type, GetCompilationDeviceType(platform_info.device_type())); XlaDeviceCompiler* xla_device_compiler = nullptr; TF_EXPECT_OK(BuildXlaDeviceCompiler(device, device_setup_.flr(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Jan 14 15:17:12 UTC 2024 - 13.6K bytes - Viewed (0) -
tensorflow/compiler/jit/get_compiler_ir.cc
auto compilation_device_type = platform_info.device_type(); if (platform_info.device_type() != DEVICE_TPU) { TF_ASSIGN_OR_RETURN(compilation_device_type, GetCompilationDeviceType(platform_info.device_type())); } XlaDeviceCompiler* xla_device_compiler; TF_RETURN_IF_ERROR(dev->resource_manager()->LookupOrCreate<XlaDeviceCompiler>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 19K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_platform_info.h
absl::string_view visible_device_list); // Returns the device type for building a DeviceCompiler from the given platform // type. absl::StatusOr<DeviceType> GetCompilationDeviceType( const DeviceType& platform_device_type); // Builds a DeviceCompiler that uses xla::LocalClient using `platform_info` and // `compilation_device_type` (in non-TPU case) and sets *xla_device_compiler to
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_on_demand_op.cc
// in the ResourceMgr. ResourceMgr* rm = ctx->resource_manager(); CHECK(rm); TF_ASSIGN_OR_RETURN(DeviceType compilation_device_type, GetCompilationDeviceType(platform_info_.device_type())); 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: Thu Feb 29 08:39:39 UTC 2024 - 13.4K bytes - Viewed (0) -
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)