- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 53 for set_device (0.23 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/colocate_tpu_copy_with_dynamic_shape.cc
if (otherDevice->device_) { return SetDevice(otherDevice->device_); } return ChangeResult::NoChange; } void print(raw_ostream &os) const override { if (device_) { os << "<" << device_ << ">"; } else { os << "<no device>"; } } ChangeResult SetDevice(mlir::StringAttr device) { bool changed = (device != device_);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 23 00:30:27 UTC 2023 - 5.2K 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/c/experimental/stream_executor/stream_executor.h
// Used for printing. Must be null-terminated. const char* pci_bus_id; } SP_Device; #define SP_DEVICE_STRUCT_SIZE TF_OFFSET_OF_END(SP_Device, pci_bus_id) typedef struct SE_CreateDeviceParams { size_t struct_size; void* ext; // reserved for future use int32_t ordinal; // device index SP_Device* device; // Input/output, struct_size set by TF for plugin to read.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 24 08:40:35 UTC 2022 - 21.6K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor_test_util.cc
SP_Event event, TF_Status* const status) {} void WaitForEvent(const SP_Device* const device, SP_Stream stream, SP_Event event, TF_Status* const status) {} void CreateTimer(const SP_Device* const device, SP_Timer* timer, TF_Status* const status) {} void DestroyTimer(const SP_Device* const device, SP_Timer timer) {}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 21 20:09:00 UTC 2023 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/cross_device_transfer.mlir
// CHECK-NEXT: %[[RESULT_3:.*]] = corert.executeop(%[[RESULT_0]]) "tf.AddV2"(%[[RESULT_2]], %[[RESULT_2]]) %3 = corert.executeop(%0) "tf.AddV2"(%2, %2) {T = f32, device = "/device:CPU:0"} : 1 // CHECK-NEXT: %[[RESULT_4:.*]] = tfrt.get_device %[[ARG_0]] {device_name = "/device:GPU:0"} // CHECK-NEXT: %[[RESULT_5:.*]] = corert.get_dst_tensor_type %[[RESULT_3]], %[[RESULT_4]] // CHECK-NEXT: %[[RESULT_6:.*]] = corert.transfer %[[RESULT_3]], %[[RESULT_4]], %[[RESULT_5]]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 10:58:25 UTC 2022 - 2K bytes - Viewed (0) -
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/device_context_test.cc
tensorflow::GetXlaDeviceFlags()->tf_xla_enable_xla_devices = true; return true; }(); class DeviceContextTest : public ::testing::Test { public: void SetDevice(const string& device_type) { auto& rollout_config = GetXlaOpsCommonFlags()->tf_xla_use_device_api; rollout_config.AllowForDeviceInXlaLaunch(DeviceType(device_type));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 3.7K 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/csi/csi_block_test.go
if err != nil { t.Fatalf("failed to setup VolumeAttachment: %v", err) } t.Log("created attachment ", attachID) stagingPath, err := csiMapper.SetUpDevice() if err != nil { t.Fatalf("mapper failed to SetupDevice: %v", err) } // Check if NodeStageVolume staged to the right path svols := csiMapper.csiClient.(*fakeCsiDriverClient).nodeClient.GetNodeStagedVolumes()
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Mon Sep 11 06:07:40 UTC 2023 - 26.5K bytes - Viewed (0)