- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for device_num (0.43 sec)
-
tensorflow/compiler/jit/xla_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: Mon May 20 21:05:42 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor.cc
OwnedTFStatus c_status(TF_NewStatus()); SP_Stream stream_handle = static_cast<CStream*>(stream)->Handle(); SP_DeviceMemoryBase device_mem = DeviceMemoryBaseToC(location); stream_executor_->mem_zero(&device_, stream_handle, &device_mem, size, c_status.get()); return StatusFromTF_Status(c_status.get()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 14 07:39:19 UTC 2024 - 27.1K bytes - Viewed (0) -
tensorflow/c/kernels.cc
} TF_StringView TF_GetDeviceName(TF_OpKernelContext* ctx) { const auto& device_name = reinterpret_cast<tensorflow::OpKernelContext*>(ctx)->device()->name(); TF_StringView device_name_sv; device_name_sv.data = device_name.data(); device_name_sv.len = device_name.length(); return device_name_sv; } #if !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K 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/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/compiler/jit/kernels/xla_ops.cc
return launch_context; } Status GetTaskName(const std::string_view device_name, std::string* task_name) { string ignored; if (!DeviceNameUtils::SplitDeviceName(device_name, task_name, &ignored)) { return errors::InvalidArgument("Unable to parse device name: ", device_name); } return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0)