- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 50 for device_num (0.14 sec)
-
tensorflow/c/kernels_test.cc
static std::unique_ptr<OpKernel> GetFakeKernel(const char* device_name, const char* op_name, const char* node_name, Status* status) { NodeDef def; def.set_op(op_name); def.set_name(node_name); def.set_device(device_name); def.add_input("input1"); def.add_input("input2");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 50.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/cross_device_transfer.mlir
%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/c/eager/c_api_test_util.h
// `device_name`. TFE_TensorHandle* CreateVarHandle(TFE_Context* ctx, const tensorflow::string& device_name, const tensorflow::string& variable_name); // Create a variable with value `value` and name `variable_name` on a device // with name `device_name`. TFE_TensorHandle* CreateVariable(TFE_Context* ctx, float value,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 17 23:43:59 UTC 2023 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_base_device.h
shape_determination_fns; Options(std::string device_name_prefix, std::string device_name, int device_ordinal, std::string compilation_device_name, std::vector<XlaShapeLayoutHelpers::ShapeDeterminationFns> shape_determination_fns) : device_name_prefix(device_name_prefix), device_name(device_name), device_ordinal(device_ordinal),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 4K bytes - Viewed (0) -
tensorflow/c/eager/immediate_execution_tensor_handle.cc
} Status s; const char* device_name = DeviceName(&s); if (!s.ok()) { device_name = "<error fetching device name>"; } return absl::StrCat("TensorHandle(", value_string, ", shape=", shape_string, ", dtype=", DataType_Name(DataType()), ", device=\"", device_name, "\")"); } Status ImmediateExecutionTensorHandle::SummarizeValue(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 2.1K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
} } // namespace extern "C" { bool TFE_IsCustomDevice(TFE_Context* ctx, const char* device_name) { return tensorflow::unwrap(ctx)->IsCustomDevice(device_name); } void TFE_RegisterCustomDevice(TFE_Context* ctx, TFE_CustomDevice device, const char* device_name, void* device_info, TF_Status* status) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 08:11:23 UTC 2024 - 44K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.h
template <std::size_t num_devices> void RegisterParallelDevice( TFE_Context* context, const char* device_name, const std::array<const char*, num_devices>& underlying_devices, TF_Status* status) { TFE_CustomDevice device; void* device_info; tensorflow::parallel_device::AllocateParallelDevice( device_name, underlying_devices.data(), underlying_devices.size(), &device, &device_info);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 09 01:12:35 UTC 2021 - 6.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/mark_initialized_variables.cc
bool IsVariableInitialized(mlir::TF::VarHandleOp var_handle_op, llvm::StringRef device_name, const tensorflow::DeviceMgr* mgr, tensorflow::Session* session) { auto* var_ptr = GetVariableFromSession(var_handle_op, device_name, mgr); if (!var_ptr) return false; auto* tensor = var_ptr->tensor();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 13 19:14:56 UTC 2023 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/jit/device_util_test.cc
for (int i = 0; i < num_devices; i++) { string device_name = absl::StrCat("/job:localhost/replica:0/task:0/device:XPU:", i); TF_ASSERT_OK_AND_ASSIGN(jit::DeviceId device_id, device_info_cache.GetIdFor(device_name)); device_set.Insert(device_id); expected_devices.push_back(device_name); } device_set.ForEach([&](jit::DeviceId device_id) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/c/eager/dlpack.cc
DLDevice GetDlContext(TFE_TensorHandle* h, TF_Status* status) { DLDevice ctx; const char* device_name = tensorflow::unwrap(h)->BackingDeviceName(&status->status); DeviceNameUtils::ParsedName parsed_name; tensorflow::DeviceNameUtils::ParseFullName(device_name, &parsed_name); std::string device_type = parsed_name.type; int device_id = 0; if (parsed_name.has_id) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 12.8K bytes - Viewed (0)