Home
last modified time | relevance | path

Searched refs:aligned_ch_count (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_constants.h118 const int aligned_ch_count = use_dot_conv ? weights.shape.o * src_depth * 4 in UploadWeightsForConvConstants() local
120 const int float_count = aligned_ch_count * kernel_x * kernel_y; in UploadWeightsForConvConstants()
Dconv_constants.cc254 const int aligned_ch_count = in IsConvConstantsSupported() local
256 const int filters_count = aligned_ch_count * w_shape.h * w_shape.w; in IsConvConstantsSupported()