- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 53 for set_device (0.18 sec)
-
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) -
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) -
tensorflow/compiler/jit/pjrt_compile_util_test.cc
TEST(PjrtCompileUtilTest, CompileToPjRtLoadedExecutable) { DeviceSetup device_setup; TF_ASSERT_OK_AND_ASSIGN(auto fdef, SampleFuntionAddXY("foo")); device_setup.AddDevicesAndSetUp({DEVICE_GPU}, fdef); Device* device = device_setup.GetDevice(DEVICE_GPU); const XlaPlatformInfo platform_info = XlaPlatformInfoFromDevice(device); NameAttrList function; function.set_name("foo"); ResourceMgr resource_mgr("");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 21 23:21:57 UTC 2023 - 4.7K 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/ir/tf_remaining_ops.cc
return getKey().str(); } namespace { std::string GetRendezvousKey(const std::string& send_device, const uint64_t send_device_incarnation, const std::string& recv_device, const std::string& tensor_name) { return absl::StrCat(send_device, ";", send_device_incarnation, ";", recv_device, ";", tensor_name); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 14 20:05:58 UTC 2023 - 7.6K 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)