- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 26 for set_device (0.44 sec)
-
tensorflow/compiler/jit/xla_platform_info_test.cc
}; #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM TEST_F(XlaPlatformInfoTest, BuildXlaDeviceCompilerXlaDeviceMetadata) { device_setup_.AddDevicesAndSetUp({DEVICE_XLA_GPU}); Device* device = device_setup_.GetDevice(DEVICE_XLA_GPU); const XlaDevice::Metadata* metadata = nullptr; TF_CHECK_OK(XlaDevice::GetMetadataFromDevice(device, &metadata)); XlaPlatformInfo platform_info = XlaPlatformInfoFromDevice(device);
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/c/eager/c_api_test_util.cc
TF_DeleteStatus(status); TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(in)); TFE_OpSetAttrString(op, "tensor_name", "dummy", 5); TFE_OpSetAttrString(op, "send_device", send_device.c_str(), send_device.size()); TFE_OpSetAttrString(op, "recv_device", recv_device.c_str(), recv_device.size()); TFE_OpSetAttrInt(op, "send_device_incarnation", send_device_incarnation);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compiler_options_util_test.cc
} DeviceSetup device_setup_; }; TEST_F(XlaCompilerOptionsTest, PjRtOptionsXlaDevice) { device_setup_.AddDevicesAndSetUp({DEVICE_XLA_GPU}); Device* device = device_setup_.GetDevice(DEVICE_XLA_GPU); DeviceType compilation_device_type = DeviceType(DEVICE_GPU_XLA_JIT); se::Platform::Id platform_id = nullptr; auto xla_device_metadata = CreateXlaDeviceMetadata(compilation_device_type);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 29 01:41:20 UTC 2023 - 14.8K bytes - Viewed (0) -
pkg/volume/volume.go
type CustomBlockVolumeMapper interface { BlockVolumeMapper // SetUpDevice prepares the volume to the node by the plugin specific way. // For most in-tree plugins, attacher.Attach() and attacher.WaitForAttach() // will do necessary works. // This may be called more than once, so implementations must be idempotent. // SetUpDevice returns stagingPath if device setup was successful SetUpDevice() (stagingPath string, err error)
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Tue May 14 06:17:25 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor.cc
SP_DEVICE_MEMORY_BASE_STRUCT_SIZE); // All other fields could theoretically be zero/null. return absl::OkStatus(); } absl::Status ValidateSPDevice(const SP_Device& device) { TF_VALIDATE_STRUCT_SIZE(SP_Device, device, SP_DEVICE_STRUCT_SIZE); // All other fields could theoretically be zero/null. return absl::OkStatus(); } absl::Status ValidateSPDeviceFns(const SP_DeviceFns& device_fns) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 14 07:39:19 UTC 2024 - 27.1K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device.h
// registered on 'jit_device'. Returns ownership of a XlaDeviceOpRegistrations // object that encapsulates the kernel registrations. struct XlaDeviceOpRegistrations { std::vector<std::unique_ptr<kernel_factory::OpKernelRegistrar>> op_kernel_registrars; }; XlaDeviceOpRegistrations* RegisterXlaDeviceKernels( const char* device, const char* jit_device,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device.cc
XlaDeviceOpRegistrations* RegisterXlaDeviceKernels( const char* device, const char* jit_device, OpKernel* (*factory)(OpKernelConstruction*), StringPiece kernel_class_name) { XlaOpRegistry::RegisterCompilationKernels(); XlaDeviceOpRegistrations* registrations = new XlaDeviceOpRegistrations; for (const KernelDef* jit_def : XlaOpRegistry::DeviceKernels( jit_device, /*include_compilation_only_kernels=*/false)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 21:05:42 UTC 2024 - 24.3K bytes - Viewed (0) -
pkg/volume/local/local_test.go
_, err = customMapper.SetUpDevice() if err != nil { t.Errorf("Failed to SetUpDevice, err: %v", err) } devPath, err = customMapper.MapPodDevice() if err != nil { t.Errorf("Failed to MapPodDevice, err: %v", err) } } if _, err := os.Stat(devPath); err != nil { if os.IsNotExist(err) { t.Errorf("SetUpDevice() failed, volume path not created: %s", devPath) } else {
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Mon May 08 10:53:39 UTC 2023 - 22.5K bytes - Viewed (0) -
platforms/core-runtime/logging/src/main/java/org/gradle/internal/deprecation/DeprecationMessageBuilder.java
void setBuildInvocationUsage() { this.usageType = DeprecatedFeatureUsage.Type.BUILD_INVOCATION; } void setSummary(String summary) { this.summary = summary; } void setAdvice(String advice) { this.advice = advice; } void setDeprecationTimeline(DeprecationTimeline deprecationTimeline) { this.deprecationTimeline = deprecationTimeline; }
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Tue May 28 13:09:37 UTC 2024 - 19.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc
rewriter.replaceOp(op_to_replace, ValueRange({fused_op})); return success(); } }; const char kDeviceAttr[] = "device"; const char kDeviceGpu[] = "GPU"; std::optional<std::string> GetDevice(mlir::Operation *op) { mlir::StringAttr device = op->getAttrOfType<mlir::StringAttr>(kDeviceAttr); if (!device || device.getValue().empty()) { return std::nullopt; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.9K bytes - Viewed (0)