Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dprecision.h26 enum class CalculationsPrecision { F32, F32_F16, F16 }; enumerator
Dprecision.cc23 case CalculationsPrecision::F32_F16: in ToString()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dfully_connected_test_util.cc132 case CalculationsPrecision::F32_F16: in FullyConnectedExtraLargeTest()
142 if (precision == CalculationsPrecision::F32_F16 && in FullyConnectedExtraLargeTest()
Dquantize_and_dequantize.cc32 definition.precision == CalculationsPrecision::F32_F16; in CreateQuantizeAndDequantize()
Dfully_connected.cc93 case CalculationsPrecision::F32_F16: in GetFullyConnectedKernelCode()
Dconvolution_transposed_thin.cc69 case CalculationsPrecision::F32_F16: in GenerateConvolutionTransposedCode()
Dconv_buffer_1x1.cc65 case CalculationsPrecision::F32_F16: in GetComputationPart()
201 case CalculationsPrecision::F32_F16: in GenerateConvBuffer1x1()
Dwinograd.cc254 case CalculationsPrecision::F32_F16: in GetWinograd4x4To36TileX6Code()
535 case CalculationsPrecision::F32_F16: in GetWinograd36To4x4Tile4x1Code()
Dconvolution_transposed_3x3_thin.cc71 case CalculationsPrecision::F32_F16: in GenerateConvolutionTransposedCode()
Dconvolution_transposed.cc181 case CalculationsPrecision::F32_F16: in GenerateConvolutionTransposedCode()
240 case CalculationsPrecision::F32_F16: in GenerateConvolutionTransposedCode()
Dconv_constants.cc67 if (precision == CalculationsPrecision::F32_F16) { in GenerateConv()
Dconvolution_transposed_4x4.cc126 case CalculationsPrecision::F32_F16: in GenerateConvolutionTransposedCode()
Dconv_metal.cc484 if (definition.precision == CalculationsPrecision::F32_F16) { in GenerateConvolution()
875 if (precision == CalculationsPrecision::F32_F16) { in GetConvParamsForIntel()
913 if (precision == CalculationsPrecision::F32_F16) { in GetConvParamsForAMD()
Dconvolution_transposed_3x3.cc106 case CalculationsPrecision::F32_F16: in GenerateConvolutionTransposedCode()
Dconv_powervr.cc793 !(op_def.precision == CalculationsPrecision::F32_F16 && in GenerateConv()
1238 } else if (definition.precision == CalculationsPrecision::F32_F16) { in GuessBestParams()
1271 if (definition.precision != CalculationsPrecision::F32_F16 && in GuessBestParams()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Denvironment.cc76 case CalculationsPrecision::F32_F16: in IsGpuSupportsPrecision()
144 {CalculationsPrecision::F32, CalculationsPrecision::F32_F16, in GetSupportedPrecisions()
Dcl_operation.cc117 case CalculationsPrecision::F32_F16: in GetCommonOpenCLDefines()
Dserialization.cc231 case CalculationsPrecision::F32_F16: in ToFB()
232 return data::CalculationsPrecision::F32_F16; in ToFB()
274 case data::CalculationsPrecision::F32_F16: in ToEnum()
275 return CalculationsPrecision::F32_F16; in ToEnum()
Dapi.cc605 precision = CalculationsPrecision::F32_F16; in GetPrecision()
616 precision = CalculationsPrecision::F32_F16; in GetPrecision()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dutil.cc114 case CalculationsPrecision::F32_F16: in GetRecommendedBlockSizeForConv()
Dserialization_base.fbs196 F32_F16 = 1,
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dtest_util.cc44 return {CalculationsPrecision::F32, CalculationsPrecision::F32_F16, in GetSupportedPrecisions()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Dfc_fc_add.cc86 case CalculationsPrecision::F32_F16: in GetFCFCAddKernelCode()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dcompute_task.cc111 if (precision == CalculationsPrecision::F32_F16) { in CompileProgram()
/external/tensorflow/tensorflow/lite/delegates/gpu/
Dmetal_delegate.mm352 precision = CalculationsPrecision::F32_F16;

12