- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 42 for device_num (0.16 sec)
-
tensorflow/c/eager/c_api_test.cc
const char* device_name = TFE_OpGetDevice(matmul, status); ASSERT_TRUE(strstr(device_name, "GPU:0") != nullptr); TFE_OpSetDevice(matmul, "CPU:0", status); ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status); device_name = TFE_OpGetDevice(matmul, status); ASSERT_TRUE(strstr(device_name, "CPU:0") != nullptr); } TFE_DeleteOp(matmul);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 03 20:50:20 UTC 2023 - 94.6K 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/c/eager/immediate_execution_context.h
//===--------------------------------------------------------------------===// virtual CustomDeviceOpHandler& GetCustomDeviceOpHandler() = 0; // Returns whether `device_name` is registered as a custom device. virtual bool IsCustomDevice(const string& device_name) = 0; // Register a custom device. It will return error is the device name is // already registered.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 06 08:34:00 UTC 2023 - 12.3K bytes - Viewed (0) -
tensorflow/compiler/jit/get_compiler_ir.cc
NameAttrList function; function.set_name(std::string{func_name}); std::string device_name; if (!platform_name.empty()) { device_name = absl::StrCat("/device:", platform_name, ":0"); } FunctionLibraryRuntime* flr = pflr->GetFLR(device_name); if (flr == nullptr) { // Use CPU as the fallback to get the `FunctionLibraryRuntime`. flr = pflr->GetFLR("/device:CPU:0");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 19K 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) -
tensorflow/compiler/jit/xla_tpu_device.cc
return absl::OkStatus(); } int device_count = platform->VisibleDeviceCount(); for (int i = 0; i < device_count; ++i) { const string device_name = absl::StrCat("/physical_device:TPU:", i); devices->push_back(device_name); } return absl::OkStatus(); } Status TpuNodeDeviceFactory::CreateDevices( const SessionOptions& session_options, const string& name_prefix,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 20.9K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_cpu_device.cc
se::PlatformManager::PlatformWithName("Host")); XlaDevice::Options options; options.platform = platform; options.device_name_prefix = name_prefix; options.device_name = DEVICE_XLA_CPU; options.device_ordinal = 0; options.compilation_device_name = DEVICE_CPU_XLA_JIT; options.use_multiple_streams = false; XlaShapeLayoutHelpers::ShapeDeterminationFns shape_representation_fns{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/python/converter_python_api.cc
const char* node_name = opdef.name().c_str(); const char* op_name = opdef.name().c_str(); const char* device_name = "CPU"; static auto fake_compute_func = [](void* kernel, TF_OpKernelContext* ctx) { }; TF_KernelBuilder* builder = TF_NewKernelBuilder(op_name, device_name, /*create_func=*/nullptr, fake_compute_func, /*delete_func=*/nullptr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 19.2K bytes - Viewed (0) -
tensorflow/c/eager/c_api.h
TF_Status* status); // Create a new TFE_TensorHandle with the same contents as 'h' but placed // in the memory of the device name 'device_name'. // If source and destination are the same device, then this creates a new handle // that shares the underlying buffer. Otherwise, it currently requires at least
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 22.8K bytes - Viewed (0)