Searched refs:is_f16 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | convolution_transposed.cc | 51 const bool is_f16 = definition.precision == CalculationsPrecision::F16; in ConvolutionTransposed() local
54 block_size_ = is_f16 ? int4(2, 1, 1, 2) : int4(2, 1, 1, 1); in ConvolutionTransposed()
56 block_size_ = is_f16 ? int4(2, 2, 1, 2) : int4(2, 2, 1, 1); in ConvolutionTransposed()
97 const bool is_f16 = definition.precision == CalculationsPrecision::F16; in ConvolutionTransposed() local
100 block_size_ = is_f16 ? int4(2, 1, 1, 2) : int4(2, 1, 1, 1); in ConvolutionTransposed()
102 block_size_ = is_f16 ? int4(2, 2, 1, 2) : int4(2, 2, 1, 1); in ConvolutionTransposed()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 1708 const bool is_f16 = input_ty.getElementType().isF16(); in GetOptimalLayout() local
1709 if (is_f16 && CanUseTensorCores(devices)) return "NHWC"; in GetOptimalLayout()
1791 const bool is_f16 = input_ty.getElementType().isF16(); in GetOptimalLayout() local
1792 if (is_f16 && CanUseTensorCores(devices)) return "NHWC"; in GetOptimalLayout()
1864 const bool is_f16 = filter_ty.getElementType().isF16(); in GetOptimalLayout() local
1865 if (is_f16 && CanUseTensorCores(devices)) return "NHWC"; in GetOptimalLayout()
2318 const bool is_f16 = x_ty.getElementType().isF16(); in GetOptimalLayout() local
2319 if (is_f16 && CanUseTensorCores(devices)) return "NHWC"; in GetOptimalLayout()
2381 const bool is_f16 = x_ty.getElementType().isF16(); in GetOptimalLayout() local
2382 if (is_f16 && CanUseTensorCores(devices)) return "NHWC"; in GetOptimalLayout()
|