- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 26 for set_device (0.2 sec)
-
tensorflow/compiler/jit/xla_launch_util_gpu_test.cc
Var* var = CreateVariable<T>(name, shape, data); ResourceMgr* rm = device_->resource_manager(); TF_ASSERT_OK(rm->Create(rm->default_container(), name, var)); ResourceHandle handle; handle.set_device(device_->name()); handle.set_container(rm->default_container()); handle.set_name(name); TypeIndex type_index = TypeIndex::Make<Var>(); handle.set_hash_code(type_index.hash_code());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 10K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc
// Target the XLA CPU/GPU backends. VLOG(2) << "Replacing with XlaLaunch"; VLOG(2) << "Device is " << launch->requested_device(); def.set_op("XlaLaunch"); def.set_device(launch->requested_device()); AddNodeAttr("Tconstants", DataTypeVector{}, &def); AddNodeAttr("Targs", arg_types, &def); AddNodeAttr("Nresources", num_variables, &def);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
// associated NodeDef is not updated. for (int i = 0, e = inst->getNumOperands(); i < e; ++i) { node_def->add_input(); } if (auto attr = inst->getAttrOfType<mlir::StringAttr>("device")) { node_def->set_device(std::string(attr.getValue())); } // Add the node debug info. TF_RETURN_IF_ERROR(ConvertLocation( inst->getLoc(), name, node_def->mutable_experimental_debug_info())); return node_def; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_test.cc
Var* var = CreateVariable<T>(name, shape, data); ResourceMgr* rm = device_->resource_manager(); TF_ASSERT_OK(rm->Create(rm->default_container(), name, var)); ResourceHandle handle; handle.set_device(device_->name()); handle.set_container(rm->default_container()); handle.set_name(name); TypeIndex type_index = TypeIndex::Make<Var>(); handle.set_hash_code(type_index.hash_code());
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/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/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) -
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)