Home
last modified time | relevance | path

Searched refs:F32_F16 (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dfully_connected_test_util.cc134 case CalculationsPrecision::F32_F16: in FullyConnectedExtraLargeTest()
141 if (precision == CalculationsPrecision::F32_F16 && in FullyConnectedExtraLargeTest()
145 if (precision == CalculationsPrecision::F32_F16 && in FullyConnectedExtraLargeTest()
Dquantize_and_dequantize.cc33 definition.precision == CalculationsPrecision::F32_F16; in CreateQuantizeAndDequantize()
Dconvolution_transposed_thin.cc69 case CalculationsPrecision::F32_F16: in GenerateConvolutionTransposedCode()
Dfully_connected.cc120 case CalculationsPrecision::F32_F16: in GetFullyConnectedKernelCode()
Dconvolution_transposed_3x3_thin.cc44 case CalculationsPrecision::F32_F16: in ConvInstr()
Dconv_constants.cc76 if (precision == CalculationsPrecision::F32_F16) { in GenerateConv()
Dconvolution_transposed.cc198 case CalculationsPrecision::F32_F16: in GenerateConvolutionTransposedCode()
257 case CalculationsPrecision::F32_F16: in GenerateConvolutionTransposedCode()
Dconvolution_transposed_4x4.cc126 case CalculationsPrecision::F32_F16: in GenerateConvolutionTransposedCode()
Dconvolution_transposed_3x3.cc104 case CalculationsPrecision::F32_F16: in GenerateConvolutionTransposedCode()
Dconv_generic.cc829 if (op_def.precision != CalculationsPrecision::F32_F16) { in GenerateConv()
1707 } else if (definition.precision == CalculationsPrecision::F32_F16) { in GuessBestParams()
1733 definition.precision != CalculationsPrecision::F32_F16 && in GuessBestParams()
1744 if (definition.precision != CalculationsPrecision::F32_F16 && in GuessBestParams()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dprecision.h26 enum class CalculationsPrecision { F32, F32_F16, F16 }; enumerator
Dprecision.cc25 case CalculationsPrecision::F32_F16: in ToString()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Denvironment.cc77 case CalculationsPrecision::F32_F16: in IsGpuSupportsPrecision()
145 {CalculationsPrecision::F32, CalculationsPrecision::F32_F16, in GetSupportedPrecisions()
Dcl_operation.cc98 case CalculationsPrecision::F32_F16: in GetCommonOpenCLDefines()
Dapi.cc617 precision = CalculationsPrecision::F32_F16; in GetPrecision()
628 precision = CalculationsPrecision::F32_F16; in GetPrecision()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dserialization_base.cc225 case CalculationsPrecision::F32_F16: in ToFB()
226 return data::CalculationsPrecision::F32_F16; in ToFB()
268 case data::CalculationsPrecision::F32_F16: in ToEnum()
269 return CalculationsPrecision::F32_F16; in ToEnum()
Dserialization_base.fbs164 F32_F16 = 1,
Dutil.cc168 case CalculationsPrecision::F32_F16: in GetRecommendedBlockSizeForConv()
Dserialization_base_generated.h285 F32_F16 = 1, enumerator
294 CalculationsPrecision::F32_F16, in EnumValuesCalculationsPrecision()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/benchmarking/
Dmain.mm130 precision = CalculationsPrecision::F32_F16;
197 precision = CalculationsPrecision::F32_F16;
263 precision = CalculationsPrecision::F32_F16;
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dtest_util.cc45 return {CalculationsPrecision::F32, CalculationsPrecision::F32_F16, in GetSupportedPrecisions()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dcompute_task.cc77 if (precision == CalculationsPrecision::F32_F16) { in GetMetalDefines()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Dfc_fc_add.cc115 case CalculationsPrecision::F32_F16: in GetFCFCAddKernelCode()
/external/tensorflow/tensorflow/lite/delegates/gpu/
Dmetal_delegate.mm352 precision = CalculationsPrecision::F32_F16;