- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for CanUseTensorCores (0.24 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_device_helper.h
bool CanUseGpuDevice(const RuntimeDevices &devices); // Returns true if all of the GPUs available at runtime support TensorCores // (NVIDIA compute capability >= 7.0). bool CanUseTensorCores(const RuntimeDevices &devices); // Returns true if operation does not have explicit device placement that would // prevent it from running on GPU device. bool CanUseGpuDevice(Operation *op); } // namespace TF
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Nov 12 21:57:12 UTC 2021 - 1.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_device_helper.cc
return llvm::any_of(devices.device_names(), IsGpuDevice); } // Returns true if all of the GPUs available at runtime support TensorCores // (NVIDIA compute capability >= 7.0). bool CanUseTensorCores(const RuntimeDevices &devices) { auto has_tensor_cores = [&](const DeviceNameUtils::ParsedName &device) { auto md = devices.GetGpuDeviceMetadata(device); return md ? md->getCcMajor() >= 7 : false; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 21 08:41:18 UTC 2022 - 2.4K bytes - Viewed (0)