- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for set_device (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
// associated NodeDef is not updated. for (int i = 0, e = inst->getNumOperands(); i < e; ++i) { node_def->add_input(); } if (auto attr = inst->getAttrOfType<mlir::StringAttr>("device")) { node_def->set_device(std::string(attr.getValue())); } // Add the node debug info. TF_RETURN_IF_ERROR(ConvertLocation( inst->getLoc(), name, node_def->mutable_experimental_debug_info())); return node_def; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
const std::string& src_device = ctx->device()->name(); std::string task_prefix; TF_RETURN_IF_ERROR(GetTaskName(src_device, &task_prefix)); const std::string dst_device = absl::StrCat(task_prefix, "/device:CPU:0"); const std::string& rendezvous_key = Rendezvous::CreateKey(src_device, /*src_incarnation=*/1, dst_device,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor_test.cc
se_.create_event = [](const SP_Device* const device, SP_Event* event, TF_Status* const status) -> void { *event = new SP_Event_st(123); }; se_.destroy_event = [](const SP_Device* const device, SP_Event event) -> void { delete event; }; se_.get_event_status = [](const SP_Device* const device,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 19:54:04 UTC 2024 - 26.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_device_propagation.cc
auto it = value_to_device.find(operand); if (it == value_to_device.end()) return llvm::StringRef(); if (new_device.empty()) { new_device = it->getSecond(); continue; } if (new_device != it->getSecond()) return llvm::StringRef(); } return new_device; } // Propagates devices from function arguments. void PropagateDevicesFromArguments( func::FuncOp func,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K 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.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) -
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) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
execute_launch.getLoc(), llvm::ArrayRef<Type>{}, reformat_operands); WrapOpInLaunch(&builder, execute_launch.getLoc(), reformat_op, execute_launch.getDevice()); // Build the replicated unformat op after the loop. First prepare building the // replicate op. llvm::SmallVector<std::pair<ValueRange, Type>, 8> unformat_replicate_operands;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0)