| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/ |
| D | convolution_transposed_selector.cc | 104 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionTransposedWithDynamicWeights()
109 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionTransposedWithDynamicWeights()
117 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionTransposedWithDynamicWeights()
123 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionTransposedWithDynamicWeights()
128 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionTransposedWithDynamicWeights()
133 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionTransposedWithDynamicWeights()
139 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionTransposedWithDynamicWeights()
|
| D | convolution_selector.cc | 65 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionDynamicWeightsAdreno()
125 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionDynamicWeightsMali()
130 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionDynamicWeightsMali()
206 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionWithDynamicWeights()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | convolution_transposed.h | 55 WeightsDescription GetWeightsDescription() const { in GetWeightsDescription() function
101 GetTotalElementsCountForLayout(GetWeightsDescription(), weights.shape); in UploadWeights()
107 RearrangeWeights(weights, GetWeightsDescription(), weights_type, in UploadWeights()
|
| D | convolution_transposed_3x3_thin.cc | 61 if (GetWeightsDescription().IsI4O4()) { in GenerateConvolutionTransposedCode()
206 GetTotalElementsCountForLayout(GetWeightsDescription(), weights.shape); in UploadWeights()
219 RearrangeWeights(weights, GetWeightsDescription(), weights_type, in UploadWeights()
|
| D | convolution_transposed_4x4.cc | 116 if (GetWeightsDescription().IsI4O4()) { in GenerateConvolutionTransposedCode()
395 GetTotalElementsCountForLayout(GetWeightsDescription(), weights.shape); in UploadWeights()
412 RearrangeWeights(weights, GetWeightsDescription(), weights_type, in UploadWeights()
|
| D | convolution_transposed_3x3_thin.h | 51 WeightsDescription GetWeightsDescription() const { in GetWeightsDescription() function
|
| D | convolution_transposed_3x3.cc | 96 if (GetWeightsDescription().IsI4O4()) { in GenerateConvolutionTransposedCode()
390 GetTotalElementsCountForLayout(GetWeightsDescription(), weights.shape); in UploadWeights()
407 RearrangeWeights(weights, GetWeightsDescription(), weights_type, in UploadWeights()
|
| D | conv_powervr.h | 50 WeightsDescription GetWeightsDescription() const { in GetWeightsDescription() function
257 GetTotalElementsCountForLayout(GetWeightsDescription(), weights.shape); in UploadWeights()
261 RearrangeWeights(weights, GetWeightsDescription(), weights_type, in UploadWeights()
|
| D | convolution_transposed_3x3.h | 54 WeightsDescription GetWeightsDescription() const { in GetWeightsDescription() function
|
| D | convolution_transposed_4x4.h | 56 WeightsDescription GetWeightsDescription() const { in GetWeightsDescription() function
|
| D | conv_metal.h | 72 WeightsDescription GetWeightsDescription() const { in GetWeightsDescription() function
|
| D | conv_buffer_1x1.h | 51 WeightsDescription GetWeightsDescription() const { in GetWeightsDescription() function
|
| D | conv_powervr.cc | 824 if (GetWeightsDescription().IsI4O4()) { in GenerateConv()
852 if (GetWeightsDescription().IsI4O4()) { in GenerateConv()
883 if (GetWeightsDescription().IsI4O4()) { in GenerateConv()
|
| D | conv_metal.cc | 1043 attr.weights, desc.GetWeightsDescription(), weights_type); in CreateConvolutionMetal()
1159 wino_weights, desc.GetWeightsDescription(), weights_type); in CreateConvolutionMetalWino4x4To6x6()
|
| D | convolution_transposed.cc | 171 if (GetWeightsDescription().IsI4O4()) { in GenerateConvolutionTransposedCode()
|