- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 32 for device_num (0.63 sec)
-
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/compiler/jit/xla_gpu_device.cc
gpu_ids->insert(i); } } for (int i : *gpu_ids) { XlaDevice::Options options; options.platform = platform.value(); options.device_name_prefix = name_prefix; options.device_name = DEVICE_XLA_GPU; options.device_ordinal = i; options.compilation_device_name = DEVICE_GPU_XLA_JIT; options.use_multiple_streams = true; options.allowed_devices = gpu_ids;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 6.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc
if (!device || device.getValue().empty()) { return std::nullopt; } const std::string device_name = device.str(); tensorflow::DeviceNameUtils::ParsedName parsed_name; if (!tensorflow::DeviceNameUtils::ParseFullName(device_name, &parsed_name)) { return std::nullopt; } if (!parsed_name.has_type) { return std::nullopt; } return parsed_name.type; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.9K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_cluster_util.cc
flags->xla_auto_jit_flag.optimization_level_general); } return result; } int GetGpuNumber(const string& device_name) { DeviceNameUtils::ParsedName parsed_name; if (!DeviceNameUtils::ParseFullName(device_name, &parsed_name)) { return -1; } return parsed_name.type == DEVICE_GPU ? parsed_name.id : -1; } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 08:39:39 UTC 2024 - 21.3K bytes - Viewed (0) -
cluster/gce/gci/configure-helper.sh
local i=0 for ssd in /dev/disk/by-id/google-local-ssd-*; do if [ -e "${ssd}" ]; then local devicenum devicenum=$(echo "${ssd}" | sed -e 's/\/dev\/disk\/by-id\/google-local-ssd-\([0-9]*\)/\1/') if [[ "${i}" -lt "${scsiblocknum}" ]]; then mount-ext "${ssd}" "${devicenum}" "scsi" "block" else # GKE does not set NODE_LOCAL_SSDS so all non-block devices
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Mon Jun 10 22:07:47 UTC 2024 - 141.1K 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) -
tensorflow/c/kernels.h
// some reason, the kernel builder will not be registered, the caller should // delete it with TF_DeleteKernelBuilder. TF_CAPI_EXPORT extern TF_KernelBuilder* TF_NewKernelBuilder( const char* op_name, const char* device_name, void* (*create_func)(TF_OpKernelConstruction*), void (*compute_func)(void*, TF_OpKernelContext*), void (*delete_func)(void*)); // Allocates a new kernel builder and returns a pointer to it. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 09 22:46:22 UTC 2024 - 24.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device.h
se::Platform* platform = nullptr; // The device name's prefix (e.g., "/task:7") string device_name_prefix; // The name of the XLA device (e.g., "XLA_CPU") string device_name; // The number of the device. int device_ordinal = -1; // The name of the compilation device (e.g., "XLA_CPU_JIT"); string compilation_device_name;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util_test.cc
llvm::SmallVectorImpl<Device>* parsed_devices) { parsed_devices->reserve(device_names.size()); for (const auto& device_name : device_names) { Device parsed_name; if (!DeviceNameUtils::ParseFullName(device_name, &parsed_name)) return false; parsed_devices->push_back(parsed_name); } return true; } using DeviceNames = llvm::SmallVector<std::string, 8>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 46.8K bytes - Viewed (0)