- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for SetDevice (0.24 sec)
-
tensorflow/compiler/jit/xla_host_send_recv_device_context_test.cc
#include "tensorflow/core/framework/tensor_testutil.h" #include "tsl/lib/core/status_test_util.h" namespace tensorflow { namespace { class XlaHostSendRecvDeviceContextTest : public ::testing::Test { public: void SetDevice(const string& device_type) { auto device_factory = DeviceFactory::GetFactory(device_type); SessionOptions options; std::vector<std::unique_ptr<Device>> devices; Status s = device_factory->CreateDevices(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_test.cc
auto device = DeviceFactory::NewDevice(device_type.type_string(), SessionOptions(), "/job:localhost/replica:0/task:0"); device_ = device.get(); SetDevice(device_type, std::move(device)); // Create PjRtClient for XLA_CPU. TF_CHECK_OK(SetPjRtClientInTFGlobalResourceManager( device_type,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 28.8K bytes - Viewed (0) -
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/compiler/mlir/tensorflow/transforms/launch_to_device_attribute.cc
return WalkResult::advance(); } else if (device_str_attr.getValue() != launch.getDevice()) { return launch.emitOpError() << "inner op has conflicting 'device' attribute, " "got '" << device_str_attr.getValue() << "' but expected '" << launch.getDevice() << "'"; } } else { return launch.emitOpError()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/cluster_util_test.cc
OwningOpRef<ModuleOp> mlir_module; auto status = tensorflow::DeserializeMlirModule(string, context, &mlir_module); if (!status.ok()) { return status; } return mlir_module; } std::string GetDevice(Operation* op) { auto device_attr = op->getAttrOfType<StringAttr>("device"); return device_attr ? device_attr.getValue().str() : ""; } bool CanBeIgnoredInCluster(Operation* op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/jit/test_util.cc
TF_GRAPH_DEF_VERSION, lib_def_.get(), opts, /*default_thread_pool=*/nullptr, /*cluster_flr=*/nullptr); flr_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); } Device* DeviceSetup::GetDevice(const string& device_name) { if (device_mgr_ == nullptr) { return nullptr; } string full_device_name = absl::StrCat( "/job:localhost/replica:0/task:0/device:", device_name, ":0"); Device* device;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 3.7K 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) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_dynamic_layout_pass.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.7K bytes - Viewed (0)