Home
last modified time | relevance | path

Searched refs:fp32_weights (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Ddepthwise_conv.h77 const bool fp32_weights = precision == CalculationsPrecision::F32; in UploadWeightsForDWConv2D() local
78 const int float4_size = fp32_weights ? 16 : 8; in UploadWeightsForDWConv2D()
82 if (fp32_weights) { in UploadWeightsForDWConv2D()
92 desc.element_type = fp32_weights ? DataType::FLOAT32 : DataType::FLOAT16; in UploadWeightsForDWConv2D()
99 desc.element_type = fp32_weights ? DataType::FLOAT32 : DataType::FLOAT16; in UploadWeightsForDWConv2D()
152 const bool fp32_weights = precision == CalculationsPrecision::F32; in UploadWeightsForDWConv3D() local
153 const int float4_size = fp32_weights ? 16 : 8; in UploadWeightsForDWConv3D()
157 if (fp32_weights) { in UploadWeightsForDWConv3D()
167 desc.element_type = fp32_weights ? DataType::FLOAT32 : DataType::FLOAT16; in UploadWeightsForDWConv3D()
175 desc.element_type = fp32_weights ? DataType::FLOAT32 : DataType::FLOAT16; in UploadWeightsForDWConv3D()
Ddepthwise_conv_3x3_stride_h2.h78 const bool fp32_weights = definition_.precision == CalculationsPrecision::F32; in UploadWeightsAndBiases() local
79 const int float4_size = fp32_weights ? 16 : 8; in UploadWeightsAndBiases()
82 if (fp32_weights) { in UploadWeightsAndBiases()
94 desc.element_type = fp32_weights ? DataType::FLOAT32 : DataType::FLOAT16; in UploadWeightsAndBiases()
102 desc.element_type = fp32_weights ? DataType::FLOAT32 : DataType::FLOAT16; in UploadWeightsAndBiases()
Ddepthwise_conv_3x3.h84 const bool fp32_weights = definition_.precision == CalculationsPrecision::F32; in UploadWeightsAndBiases() local
85 const int float4_size = fp32_weights ? 16 : 8; in UploadWeightsAndBiases()
88 if (fp32_weights) { in UploadWeightsAndBiases()
100 desc.element_type = fp32_weights ? DataType::FLOAT32 : DataType::FLOAT16; in UploadWeightsAndBiases()
108 desc.element_type = fp32_weights ? DataType::FLOAT32 : DataType::FLOAT16; in UploadWeightsAndBiases()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Ddepthwise_conv_plus_1x1_conv.cc92 const bool fp32_weights = precision == CalculationsPrecision::F32; in UploadWeights() local
93 const int float_size = fp32_weights ? 4 : 2; in UploadWeights()
95 desc.element_type = fp32_weights ? DataType::FLOAT32 : DataType::FLOAT16; in UploadWeights()
101 if (fp32_weights) { in UploadWeights()