Searched refs:use_dot_conv (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_constants.cc | 51 std::string GenerateConv(int src_size, int dst_size, bool use_dot_conv, in GenerateConv() argument
56 if (use_dot_conv) { in GenerateConv()
100 bool use_dot_conv, in GenerateConvolutionConstantCode() argument
198 c += GenerateConv(src_ch_count, dst_ch_count, use_dot_conv, in GenerateConvolutionConstantCode()
201 filters_counter += use_dot_conv ? dst_ch_count : src_ch_count; in GenerateConvolutionConstantCode()
249 const bool use_dot_conv = in IsConvConstantsSupported() local
255 use_dot_conv ? w_shape.o * src_depth * 4 : w_shape.i * dst_depth * 4; in IsConvConstantsSupported()
269 const bool use_dot_conv = in CreateConvConstants() local
273 use_dot_conv, &op); in CreateConvConstants()
286 definition, attr.weights.shape, stride_correction, use_dot_conv, &op); in CreateConvConstants()
|
| D | conv_constants.h | 110 bool use_dot_conv, GPUOperation* op) { in UploadWeightsForConvConstants() argument
118 const int aligned_ch_count = use_dot_conv ? weights.shape.o * src_depth * 4 in UploadWeightsForConvConstants()
131 if (use_dot_conv) { in UploadWeightsForConvConstants()
140 if (use_dot_conv) { in UploadWeightsForConvConstants()
|