Home
last modified time | relevance | path

Searched refs:CalculationsPrecision (Results 1 – 25 of 86) sorted by relevance

1234

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dprecision.cc21 std::string ToString(CalculationsPrecision precision) { in ToString()
23 case CalculationsPrecision::F32_F16: in ToString()
25 case CalculationsPrecision::F32: in ToString()
27 case CalculationsPrecision::F16: in ToString()
32 DataType DeduceDataTypeFromPrecision(CalculationsPrecision precision) { in DeduceDataTypeFromPrecision()
33 if (precision == CalculationsPrecision::F32) { in DeduceDataTypeFromPrecision()
Dprecision.h26 enum class CalculationsPrecision { F32, F32_F16, F16 }; enum
34 DataType DeduceDataTypeFromPrecision(CalculationsPrecision precision);
36 std::string ToString(CalculationsPrecision precision);
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Denvironment.cc74 CalculationsPrecision precision) { in IsGpuSupportsPrecision()
76 case CalculationsPrecision::F32_F16: in IsGpuSupportsPrecision()
77 case CalculationsPrecision::F16: in IsGpuSupportsPrecision()
79 case CalculationsPrecision::F32: in IsGpuSupportsPrecision()
141 std::vector<CalculationsPrecision> Environment::GetSupportedPrecisions() const { in GetSupportedPrecisions()
142 std::vector<CalculationsPrecision> precisions; in GetSupportedPrecisions()
143 for (CalculationsPrecision precision : in GetSupportedPrecisions()
144 {CalculationsPrecision::F32, CalculationsPrecision::F32_F16, in GetSupportedPrecisions()
145 CalculationsPrecision::F16}) { in GetSupportedPrecisions()
153 bool Environment::IsSupported(CalculationsPrecision precision) const { in IsSupported()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dfully_connected_test_util.cc42 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in FullyConnectedTest()
88 const float eps = precision == CalculationsPrecision::F32 ? 0.0f : 1.0f; in FullyConnectedLargeTest()
129 case CalculationsPrecision::F32: in FullyConnectedExtraLargeTest()
132 case CalculationsPrecision::F32_F16: in FullyConnectedExtraLargeTest()
135 case CalculationsPrecision::F16: in FullyConnectedExtraLargeTest()
142 if (precision == CalculationsPrecision::F32_F16 && in FullyConnectedExtraLargeTest()
Delementwise_test_util.cc61 const float eps = precision == CalculationsPrecision::F32 ? 5e-5f : 1e-3f; in CosTest()
112 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-2f; in EluTest()
140 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-2f; in ExpTest()
168 const float eps = precision == CalculationsPrecision::F32 ? 1e-5f : 1e-2f; in HardSwishTest()
195 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in LogTest()
222 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in NegTest()
248 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in RsqrtTest()
277 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in SigmoidTest()
303 const float eps = precision == CalculationsPrecision::F32 ? 1e-5f : 5e-3f; in SinTest()
330 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in SqrtTest()
[all …]
Dpadding_test_util.cc39 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in PaddingAppendWidthTest()
69 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in PaddingPrependWidthTest()
99 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in PaddingAppendHeightTest()
128 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in PaddingPrependHeightTest()
157 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in PaddingAppendChannelsTest()
186 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in PaddingPrependChannelsTest()
215 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in PaddingPrependChannelsX4Test()
244 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in PaddingComplexTest()
277 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in PaddingReflectWidthTest()
307 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in PaddingReflectChannelsTest()
Dquantize_and_dequantize.cc31 const bool is_fp16 = definition.precision == CalculationsPrecision::F16 || in CreateQuantizeAndDequantize()
32 definition.precision == CalculationsPrecision::F32_F16; in CreateQuantizeAndDequantize()
42 if (definition.precision == CalculationsPrecision::F32) { in CreateQuantizeAndDequantize()
Dresize_test_util.cc40 const float eps = precision == CalculationsPrecision::F32 ? 1e-5f : 1e-2f; in ResizeBilinearAlignedTest()
72 const float eps = precision == CalculationsPrecision::F32 ? 1e-5f : 1e-2f; in ResizeBilinearNonAlignedTest()
105 const float eps = precision == CalculationsPrecision::F32 ? 1e-5f : 1e-2f; in ResizeBilinearWithoutHalfPixelTest()
137 const float eps = precision == CalculationsPrecision::F32 ? 1e-5f : 1e-2f; in ResizeBilinearWithHalfPixelTest()
169 const float eps = precision == CalculationsPrecision::F32 ? 1e-5f : 1e-2f; in ResizeNearestTest()
201 const float eps = precision == CalculationsPrecision::F32 ? 1e-5f : 1e-2f; in ResizeNearestAlignCornersTest()
233 const float eps = precision == CalculationsPrecision::F32 ? 1e-5f : 1e-2f; in ResizeNearestHalfPixelCentersTest()
Dconv_buffer_1x1.cc36 CalculationsPrecision precision) { in GetComputationPart()
54 case CalculationsPrecision::F32: in GetComputationPart()
55 case CalculationsPrecision::F16: in GetComputationPart()
65 case CalculationsPrecision::F32_F16: in GetComputationPart()
91 definition.precision != CalculationsPrecision::F32; in GetBestParams()
97 if (definition.precision == CalculationsPrecision::F16 || !can_use_flt8) { in GetBestParams()
117 if (definition.precision == CalculationsPrecision::F32 && dst_depth < 32) { in GetBestParams()
139 if (gpu_info.IsMali() && definition.precision == CalculationsPrecision::F16 && in GetBestParams()
197 case CalculationsPrecision::F32: in GenerateConvBuffer1x1()
201 case CalculationsPrecision::F32_F16: in GenerateConvBuffer1x1()
[all …]
Drelu_test_util.cc39 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in ReLUNoClipNoAlphaTest()
68 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in ReLUClipTest()
97 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in ReLUAlphaTest()
126 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in ReLUAlphaClipTest()
Dreduce_test_util.cc36 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-2f; in MeanHWTest()
63 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-2f; in ReduceSumChannelsTest()
90 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-2f; in ReduceProductChannelsTest()
118 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-2f; in ReduceMaxChannelsTest()
146 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-2f; in ReduceMinChannelsTest()
Dwinograd.cc237 if (definition_.precision == CalculationsPrecision::F16 && in Winograd4x4To36TileX6()
253 case CalculationsPrecision::F32: in GetWinograd4x4To36TileX6Code()
254 case CalculationsPrecision::F32_F16: in GetWinograd4x4To36TileX6Code()
257 case CalculationsPrecision::F16: in GetWinograd4x4To36TileX6Code()
262 const DataType accum_type = op_def.precision == CalculationsPrecision::F16 in GetWinograd4x4To36TileX6Code()
522 if (definition_.precision == CalculationsPrecision::F16 && in Winograd36To4x4Tile4x1()
534 case CalculationsPrecision::F32: in GetWinograd36To4x4Tile4x1Code()
535 case CalculationsPrecision::F32_F16: in GetWinograd36To4x4Tile4x1Code()
538 case CalculationsPrecision::F16: in GetWinograd36To4x4Tile4x1Code()
543 const DataType accum_type = op_def.precision == CalculationsPrecision::F16 in GetWinograd36To4x4Tile4x1Code()
Dquantize_and_dequantize_test_util.cc44 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-2f; in QuantAndDequant_Dim2Bits8Test()
81 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-2f; in QuantAndDequant_Dim3Bits8_NegativeRangeTest()
117 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in QuantAndDequant_Dim3Bits16Test()
154 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-2f; in QuantAndDequant_Dim2Bits16_NegativeRangeTest()
Dconv_constants.cc52 int const_mem_offset, CalculationsPrecision precision, in GenerateConv()
67 if (precision == CalculationsPrecision::F32_F16) { in GenerateConv()
243 if (gpu_info.IsAMD() && definition.precision != CalculationsPrecision::F32 && in IsConvConstantsSupported()
257 const int float_size = definition.precision == CalculationsPrecision::F32 in IsConvConstantsSupported()
287 if (definition.precision == CalculationsPrecision::F16 && in CreateConvConstants()
291 if (definition.precision != CalculationsPrecision::F32 && in CreateConvConstants()
Dfully_connected.cc90 case CalculationsPrecision::F32: in GetFullyConnectedKernelCode()
93 case CalculationsPrecision::F32_F16: in GetFullyConnectedKernelCode()
94 case CalculationsPrecision::F16: in GetFullyConnectedKernelCode()
Drelu.cc31 if (definition.precision == CalculationsPrecision::F32) { in CreateReLU()
40 if (definition.precision == CalculationsPrecision::F32) { in CreateReLU()
Dconvolution_transposed_thin.cc34 if (definition_.precision == CalculationsPrecision::F16 && in ConvolutionTransposedThin()
68 case CalculationsPrecision::F32: in GenerateConvolutionTransposedCode()
69 case CalculationsPrecision::F32_F16: in GenerateConvolutionTransposedCode()
72 case CalculationsPrecision::F16: in GenerateConvolutionTransposedCode()
Dpooling_test_util.cc42 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in AveragePoolingTest()
73 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in AveragePoolingNonEmptyPaddingTest()
105 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in MaxPoolingTest()
137 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in MaxPoolingIndicesTest()
Dsoftmax_test_util.cc37 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in SoftmaxTest()
77 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in SoftmaxBigNumberTest()
107 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in Softmax1x1Test()
146 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in Softmax1x1BigNumberTest()
Dconv_powervr_test_util.cc45 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in ConvPowerVR1x1SimpleWeightsTest()
82 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in ConvPowerVR1x1Test()
119 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in ConvPowerVRSimpleWeightsTest()
156 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in ConvPowerVRTest()
Dprelu_test_util.cc42 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in PReLUAlphaTest()
74 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in PReLUAlphaClipTest()
106 const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f; in PReLUHWCAlphaTest()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dutil.cc83 CalculationsPrecision precision, in GetRecommendedBlockSizeForConv()
96 case CalculationsPrecision::F16: in GetRecommendedBlockSizeForConv()
114 case CalculationsPrecision::F32_F16: in GetRecommendedBlockSizeForConv()
129 case CalculationsPrecision::F32: in GetRecommendedBlockSizeForConv()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dtest_util.cc42 std::vector<CalculationsPrecision>
44 return {CalculationsPrecision::F32, CalculationsPrecision::F32_F16, in GetSupportedPrecisions()
45 CalculationsPrecision::F16}; in GetSupportedPrecisions()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/
Dperformance_profiling.cc43 create_info.precision = env.IsSupported(CalculationsPrecision::F16) in RunModelSample()
44 ? CalculationsPrecision::F16 in RunModelSample()
45 : CalculationsPrecision::F32; in RunModelSample()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Dfc_fc_add.cc83 case CalculationsPrecision::F32: in GetFCFCAddKernelCode()
86 case CalculationsPrecision::F32_F16: in GetFCFCAddKernelCode()
87 case CalculationsPrecision::F16: in GetFCFCAddKernelCode()

1234