- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for CanUseTensorCores (0.11 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
// For f16 data type on devices with Tensor Cores support NHWC data format // is up to ~2x faster. const bool is_f16 = input_ty.getElementType().isF16(); if (is_f16 && CanUseTensorCores(devices)) return "NHWC"; // For f32/f16 data type decision depends on the filter size in spatial // dimensions, for other data types we keep current data format.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0)