- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 45 for set_device (0.15 sec)
-
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/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) -
tensorflow/c/eager/c_api_test_util.h
// `send_device`, `recv_device`, and `send_device_incarnation` set. TFE_Op* SendOp(TFE_Context* ctx, TFE_TensorHandle* in, const std::string& op_name, const std::string& send_device, const std::string& recv_device, tensorflow::uint64 send_device_incarnation); // Return a RecvOp op `op_name` with the attributes `send_device`,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 17 23:43:59 UTC 2023 - 7.7K bytes - Viewed (0) -
pkg/volume/csi/csi_block.go
return "", err } return publishPath, nil } // SetUpDevice ensures the device is attached returns path where the device is located. func (m *csiBlockMapper) SetUpDevice() (string, error) { klog.V(4).Infof(log("blockMapper.SetUpDevice called")) // Get csiSource from spec if m.spec == nil { return "", errors.New(log("blockMapper.SetUpDevice spec is nil")) } csiSource, err := getCSISourceFromSpec(m.spec)
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Mon Sep 11 06:07:40 UTC 2023 - 20.1K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor_internal.h
event_handle_ = nullptr; } } SP_Event Handle() { return event_handle_; } private: SP_Device* device_; SP_StreamExecutor* stream_executor_; SP_Event event_handle_; }; class CStream : public StreamCommon { public: CStream(SP_Device* device, SP_StreamExecutor* stream_executor, StreamExecutor* executor) : StreamCommon(executor), device_(device),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 14 07:39:19 UTC 2024 - 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/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)