Home
last modified time | relevance | path

Searched refs:f32_weights (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dfully_connected.h154 const bool f32_weights = definition_.precision == CalculationsPrecision::F32; in UploadWeights() local
156 const int float4_size = f32_weights ? 16 : 8; in UploadWeights()
160 desc.element_type = f32_weights ? DataType::FLOAT32 : DataType::FLOAT16; in UploadWeights()
165 if (f32_weights) { in UploadWeights()
177 desc.element_type = f32_weights ? DataType::FLOAT32 : DataType::FLOAT16; in UploadWeights()
181 if (f32_weights) { in UploadWeights()
Dconvolution_transposed_thin.h74 const bool f32_weights = definition_.precision == CalculationsPrecision::F32; in UploadData() local
75 const int flt4_size = f32_weights ? sizeof(float4) : sizeof(half4); in UploadData()
78 desc.element_type = f32_weights ? DataType::FLOAT32 : DataType::FLOAT16; in UploadData()
84 if (f32_weights) { in UploadData()
Dconvolution_transposed.h152 const bool f32_weights = definition_.precision == CalculationsPrecision::F32; in UploadWeights() local
154 const int float4_size = f32_weights ? 16 : 8; in UploadWeights()
157 if (f32_weights) { in UploadWeights()
179 desc.element_type = f32_weights ? DataType::FLOAT32 : DataType::FLOAT16; in UploadWeights()
191 desc.element_type = f32_weights ? DataType::FLOAT32 : DataType::FLOAT16; in UploadWeights()
Dconv_constants.h116 const bool f32_weights = precision == CalculationsPrecision::F32; in UploadWeightsForConvConstants() local
117 const int float_size = f32_weights ? 4 : 2; in UploadWeightsForConvConstants()
123 desc.element_type = f32_weights ? DataType::FLOAT32 : DataType::FLOAT16; in UploadWeightsForConvConstants()
129 if (f32_weights) { in UploadWeightsForConvConstants()
Dconv_buffer_1x1.h130 const bool f32_weights = definition_.precision == CalculationsPrecision::F32; in UploadWeights() local
131 const int float4_size = f32_weights ? sizeof(float4) : sizeof(half4); in UploadWeights()
138 desc.element_type = f32_weights ? DataType::FLOAT32 : DataType::FLOAT16; in UploadWeights()
144 if (f32_weights) { in UploadWeights()
Dconv_powervr.h307 const bool f32_weights = definition_.precision == CalculationsPrecision::F32; in UploadWeights() local
309 const int float4_size = f32_weights ? 16 : 8; in UploadWeights()
313 if (f32_weights) { in UploadWeights()
335 desc.element_type = f32_weights ? DataType::FLOAT32 : DataType::FLOAT16; in UploadWeights()
348 desc.element_type = f32_weights ? DataType::FLOAT32 : DataType::FLOAT16; in UploadWeights()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Dfc_fc_add.h131 const bool f32_weights = definition_.precision == CalculationsPrecision::F32; in UploadWeights() local
133 const int float4_size = f32_weights ? 16 : 8; in UploadWeights()
137 desc.element_type = f32_weights ? DataType::FLOAT32 : DataType::FLOAT16; in UploadWeights()
142 if (f32_weights) { in UploadWeights()
153 desc.element_type = f32_weights ? DataType::FLOAT32 : DataType::FLOAT16; in UploadWeights()
161 if (f32_weights) { in UploadWeights()
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dlayer_correctness_test.py206 f32_weights = f32_model.get_weights()
207 mp_model.set_weights(f32_weights)
208 distributed_mp_model.set_weights(f32_weights)
234 f32_weights = f32_model.get_weights()
236 mp_model.get_weights(), f32_weights, rtol=rtol, atol=atol)
238 distributed_mp_model.get_weights(), f32_weights, rtol=rtol, atol=atol)