Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_powervr.h52 desc.layout = conv_params_.weights_layout; in GetWeightsDescription()
53 desc.output_group_size = conv_params_.block_size.w; in GetWeightsDescription()
202 ConvParams conv_params_; variable
227 desc.element_type = conv_params_.weights_data_type; in UploadBias()
229 desc.memory_type = conv_params_.weights_upload_type == in UploadBias()
233 const int float_size = conv_params_.weights_data_type == DataType::FLOAT32 in UploadBias()
236 int aligned_channels = AlignByN(bias.shape.v, 4 * conv_params_.block_size.w); in UploadBias()
239 if (conv_params_.weights_data_type == DataType::FLOAT32) { in UploadBias()
258 DataType weights_type = conv_params_.weights_data_type; in UploadWeights()
264 if (conv_params_.AreWeightsBuffer()) { in UploadWeights()
[all …]
Dconv_powervr.cc178 conv_params_(GuessBestParams(gpu_info, definition, attr, dst_shape)) {} in ConvPowerVR()
189 conv_params_(GuessBestParams(gpu_info, definition, attr, weights_shape, in ConvPowerVR()
200 conv_params_(GuessBestParams(gpu_info, definition, attr, dst_shape)) {} in ConvPowerVR()
215 conv_params_(operation.conv_params_) {} in ConvPowerVR()
227 conv_params_(GuessBestParams(gpu_info, definition, attr, dst_shape)) {} in ConvPowerVR()
235 std::swap(conv_params_, operation.conv_params_); in operator =()
242 if (conv_params_.linear_all) { in GenerateCode()
244 } else if (conv_params_.linear_spatial) { in GenerateCode()
249 code_ = GenerateConv(gpu_info, definition_, stride_correction, conv_params_); in GenerateCode()
254 if (conv_params_.IsPrivateMemBroadcast() && gpu_info.IsCL20OrHigher()) { in GenerateCode()
[all …]
Dconv_buffer_1x1.cc150 : GPUOperation(definition), conv_params_(conv_params) { in ConvBuffer1x1()
151 code_ = GenerateConvBuffer1x1(definition_, conv_params_, &args_); in ConvBuffer1x1()
157 conv_params_(std::move(operation.conv_params_)) {} in ConvBuffer1x1()
161 std::swap(conv_params_, operation.conv_params_); in operator =()
174 if (conv_params_.element_size == 8) { in GenerateConvBuffer1x1()
176 } else if (conv_params_.element_size == 16) { in GenerateConvBuffer1x1()
304 dst_[0]->Width() * dst_[0]->Batch(), (conv_params_.element_size / 4)); in GetGridSize()
306 DivideRoundUp(dst_width_elements, conv_params_.block_size.x); in GetGridSize()
308 DivideRoundUp(dst_[0]->Height(), conv_params_.block_size.y); in GetGridSize()
310 DivideRoundUp(dst_[0]->Slices(), conv_params_.block_size.z); in GetGridSize()
Dconv_buffer_1x1.h54 desc.output_group_size = conv_params_.block_size.z; in GetWeightsDescription()
103 ConvParams conv_params_; variable
133 const int dst_depth_aligned = AlignByN(dst_depth, conv_params_.block_size.z); in UploadWeights()
146 RearrangeWeightsToOHWIOGroupI4O4(weights, conv_params_.block_size.z, in UploadWeights()
150 RearrangeWeightsToOHWIOGroupI4O4(weights, conv_params_.block_size.z, in UploadWeights()
163 int depth = AlignByN(biases.shape.v, 4 * conv_params_.block_size.z) / 4; in UploadBiases()