- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for device_num (0.23 sec)
-
tensorflow/compiler/jit/pjrt_base_device.cc
const string& device_name, int device_ordinal) { return Device::BuildDeviceAttributes( absl::StrCat(name_prefix, "/device:", device_name, ":", device_ordinal), DeviceType(device_name), Bytes(16ULL << 30), DeviceLocality(), absl::StrCat("device: ", device_name, " device")); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/jit/test_util.cc
const std::optional<FunctionDef>& fdef) { SessionOptions options; auto* device_count = options.config.mutable_device_count(); for (const auto& device_name : device_names) { device_count->insert({device_name, 1}); } std::vector<std::unique_ptr<Device>> devices; TF_CHECK_OK(DeviceFactory::AddDevices( options, "/job:localhost/replica:0/task:0", &devices));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 3.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/execution_metadata_exporter.cc
for (auto& func : *funcs) { func.walk([&hardware_names, &index](mlir::Operation* op) { auto device_name = GetDeviceName(op); if (!device_name.has_value()) return; auto iter = hardware_names->find(*device_name); if (iter == hardware_names->end()) { hardware_names->insert({*device_name, index++}); } }); } // Build the flatbuffer.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 06:11:34 UTC 2024 - 7.5K 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/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/compiler/mlir/tensorflow/utils/session_utils.cc
} tensorflow::Var* GetVariableFromSession(mlir::TF::VarHandleOp var_handle_op, llvm::StringRef device_name, const tensorflow::DeviceMgr* mgr) { tensorflow::Device* device = nullptr; if (!mgr || !mgr->LookupDevice(StringRefToView(device_name), &device).ok()) return nullptr; tensorflow::Var* var_ptr = nullptr; const auto& container = var_handle_op.getContainer().str();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/jit/test_util.h
class DeviceSetup { public: void AddDevicesAndSetUp( const std::vector<std::string>& device_names, const std::optional<FunctionDef>& fdef = std::nullopt); Device* GetDevice(const string& device_name); FunctionLibraryRuntime* flr() { return flr_; } private: FunctionLibraryRuntime* flr_; std::unique_ptr<DeviceMgr> device_mgr_; std::unique_ptr<FunctionLibraryDefinition> lib_def_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 16 18:03:15 UTC 2023 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/jit/device_util.h
} using DeviceTypeConstRef = std::reference_wrapper<const DeviceType>; absl::StatusOr<DeviceTypeConstRef> GetDeviceTypeFor( absl::string_view device_name) { TF_ASSIGN_OR_RETURN(DeviceId device_id, GetIdFor(device_name)); return std::cref(*id_to_device_type_[device_id.id()]); } string DebugString(const DeviceSet& device_set) const; private:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 17:18:31 UTC 2024 - 7.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util.cc
const std::string attrKey = "T"; for (const Node* node : graph.nodes()) { if (node->type_string() == nodeType) { auto device_name = node->assigned_device_name(); DeviceNameUtils::ParsedName device; if (DeviceNameUtils::ParseFullName(device_name, &device) && device.has_job && device.job == jobType) { for (const auto& attr : node->attrs()) { auto attr_key = attr.first;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 12:22:33 UTC 2024 - 8.9K bytes - Viewed (0)