Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dweights_conversion.cc27 uint o_aligned = AlignByN(shape.o, 4 * weight_desc.output_group_size); in GetTotalElementsCountForLayout()
49 dst_weight_desc.output_group_size, in RearrangeWeights()
54 dst_weight_desc.output_group_size, in RearrangeWeights()
62 dst_weight_desc.output_group_size, in RearrangeWeights()
67 dst_weight_desc.output_group_size, in RearrangeWeights()
102 dst_weight_desc.output_group_size, in RearrangeWeights()
107 dst_weight_desc.output_group_size, in RearrangeWeights()
116 dst_weight_desc.output_group_size, in RearrangeWeights()
121 dst_weight_desc.output_group_size, in RearrangeWeights()
Dweights_layout.h37 int output_group_size; // OGroup size member
Dweights_layout.cc26 return output_group_size; in GetOutputGroupSize()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_weights_converter_test_util.cc64 weight_desc.output_group_size); in ConvolutionWeightsConverterTest()
112 conv_weight_desc.output_group_size = 4; in ConverterToConvWeights1x1OutX4Test()
146 conv_weight_desc.output_group_size = 4; in ConverterToConvWeights1x1OutX4UnalignedTest()
179 conv_weight_desc.output_group_size = 2; in ConverterToConvWeights1x1OutX2Test()
212 conv_weight_desc.output_group_size = 2; in ConverterToConvWeightsOutX2Test()
281 conv_weight_desc.output_group_size = 4; in ConverterToConvWeights4xTexturesTest()
Dconv_metal.h75 desc.output_group_size = params_.block_size.z; in GetWeightsDescription()
Dconv_buffer_1x1.h54 desc.output_group_size = conv_params_.block_size.z; in GetWeightsDescription()
Dconvolution_transposed.h58 desc.output_group_size = block_size_.w; in GetWeightsDescription()
Dconv_powervr.h53 desc.output_group_size = conv_params_.block_size.w; in GetWeightsDescription()