- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 27 for platform_id (0.3 sec)
-
tensorflow/compiler/jit/xla_compiler_options_util_test.cc
DeviceType compilation_device_type = DeviceType(DEVICE_GPU_XLA_JIT); se::Platform::Id platform_id = nullptr; auto xla_device_metadata = CreateXlaDeviceMetadata(compilation_device_type); std::shared_ptr<se::DeviceMemoryAllocator> custom_allocator; XlaPlatformInfo platform_info( compilation_device_type, platform_id, xla_device_metadata.get(), /*pjrt_device_metadata=*/nullptr, custom_allocator);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 29 01:41:20 UTC 2023 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_platform_info.cc
persistor_config, DeviceType(DEVICE_TPU_XLA_JIT), nullptr); return absl::OkStatus(); } if (platform_info.platform_id() == nullptr) { return errors::InvalidArgument("platform_id is null."); } auto platform = se::PlatformManager::PlatformWithId(platform_info.platform_id()); if (!platform.ok()) { return platform.status(); } absl::StatusOr<xla::Compiler*> compiler_for_platform =
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/xla_platform_info.h
explicit XlaPlatformInfo( const DeviceType device_type, se::Platform::Id platform_id, const XlaDevice::Metadata* xla_device_metadata, const PjRtBaseDevice::Metadata* pjrt_device_metadata, std::shared_ptr<se::DeviceMemoryAllocator> device_allocator) : device_type_(device_type), platform_id_(platform_id), xla_device_metadata_(xla_device_metadata),
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_platform_info_test.cc
// XlaPlatformInfo::device_type() is needed to build the appropriate // XlaDeviceCompiler. Device* device = nullptr; XlaPlatformInfo platform_info(DeviceType(DEVICE_TPU), /*platform_id=*/nullptr, /*xla_device_metadata=*/nullptr, /*pjrt_device_metadata=*/nullptr, /*device_allocator=*/nullptr);
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
compiler_arg_source)); se::Platform::Id platform_id = nullptr; if (platform_name == DEVICE_CPU) { platform_id = se::host::kHostPlatformId; } XlaPlatformInfo platform_info(DeviceType(platform_name), platform_id, /*xla_device_metadata=*/nullptr, /*pjrt_device_metadata=*/nullptr,
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_compiler_options_util.cc
options.flib_def = function_library.GetFunctionLibraryDefinition(); 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 =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util.cc
se::Platform::Id XlaPlatformInfoFromDevice(DeviceBase* device_base) { auto device = static_cast<Device*>(device_base); se::Platform::Id platform_id = nullptr; if (device->device_type() == DEVICE_CPU) { platform_id = se::host::kHostPlatformId; } return platform_id; } absl::flat_hash_map<int, int> CreateVariableLookup( const std::vector<VariableInfo>& variables) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
run_options.set_rng_seed(GetXLARandomSeed()); absl::StatusOr<xla::ExecutionOutput> execution_output; bool run_synchronous = !stream || platform_info.platform_id() == se::host::kHostPlatformId; if (run_synchronous) { execution_output = executable->Run(std::move(execution_inputs), run_options); } else { execution_output =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0) -
platforms/software/dependency-management/src/main/java/org/gradle/api/internal/artifacts/ivyservice/resolveengine/graph/builder/LenientPlatformDependencyMetadata.java
private final ComponentIdentifier platformId; // just for reporting private final boolean force; private final boolean transitive; LenientPlatformDependencyMetadata(ResolveState resolveState, NodeState from, ModuleComponentSelector cs, ModuleComponentIdentifier componentId, @Nullable ComponentIdentifier platformId, boolean force, boolean transitive) { this.resolveState = resolveState;
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Tue Dec 19 15:01:08 UTC 2023 - 5.5K bytes - Viewed (0) -
platforms/software/dependency-management/src/main/java/org/gradle/api/internal/artifacts/ivyservice/resolveengine/graph/builder/LenientPlatformGraphResolveState.java
ComponentIdentifier platformId = platformState.getSelectedPlatformId(); if (platformId == null) { // Not sure this can happen, unless in error state platformId = this.platformId; } if (!componentVersion.equals(target)) {
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Wed May 15 17:38:42 UTC 2024 - 7.7K bytes - Viewed (0)