Searched refs:s_ch (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | weights_conversion.h | 52 const int s_ch = s * 4 + j; in RearrangeWeightsToOHWIOGroupI4O4() local
54 if (s_ch < weights.shape.i && d_ch < weights.shape.o) { in RearrangeWeightsToOHWIOGroupI4O4()
56 weights.shape.LinearIndex({d_ch, y, x, s_ch}); in RearrangeWeightsToOHWIOGroupI4O4()
88 const int s_ch = s * 4 + i; in RearrangeWeightsToOHWIOGroupO4I4() local
90 if (s_ch < weights.shape.i && d_ch < weights.shape.o) { in RearrangeWeightsToOHWIOGroupO4I4()
92 weights.shape.LinearIndex({d_ch, y, x, s_ch}); in RearrangeWeightsToOHWIOGroupO4I4()
125 const int s_ch = s * 4 + j; in RearrangeWeightsToODHWIOGroupI4O4() local
127 if (s_ch < weights.shape.i && d_ch < weights.shape.o) { in RearrangeWeightsToODHWIOGroupI4O4()
129 weights.shape.LinearIndex({d_ch, y, x, z, s_ch}); in RearrangeWeightsToODHWIOGroupI4O4()
162 const int s_ch = s * 4 + j; in RearrangeWeightsToI4HWIOOGroupO4() local
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_constants.h | 50 const int s_ch = s * 4 + j; in RearrangeWeightsForConvConstants() local
52 if (s_ch < weights.shape.i && d_ch < weights.shape.o) { in RearrangeWeightsForConvConstants()
54 weights.shape.LinearIndex({d_ch, y, x, s_ch}); in RearrangeWeightsForConvConstants()
87 const int s_ch = s * 4 + i; in RearrangeWeightsForConvConstantsDot() local
89 if (s_ch < weights.shape.i && d_ch < weights.shape.o) { in RearrangeWeightsForConvConstantsDot()
91 weights.shape.LinearIndex({d_ch, y, x, s_ch}); in RearrangeWeightsForConvConstantsDot()
|
| D | depthwise_conv_3x3_stride_h2.h | 122 const int s_ch = s * 4 + i; in RearrangeWeightsAndBiasesData() local
123 if (s_ch < weights.shape.i) { in RearrangeWeightsAndBiasesData()
124 const int f_index = weights.shape.LinearIndex({0, y, x, s_ch}); in RearrangeWeightsAndBiasesData()
|
| D | convolution_transposed_thin.h | 120 const int s_ch = s * 4 + i; in RearrangeWeightsData() local
122 if (s_ch < weights.shape.i && d_ch < weights.shape.o) { in RearrangeWeightsData()
123 const int f_index = weights.shape.LinearIndex({d_ch, y, x, s_ch}); in RearrangeWeightsData()
|
| D | depthwise_conv_3x3.h | 128 const int s_ch = s * 4 + i; in RearrangeWeightsAndBiasesData() local
129 if (s_ch < weights.shape.i) { in RearrangeWeightsAndBiasesData()
130 const int f_index = weights.shape.LinearIndex({0, y, x, s_ch}); in RearrangeWeightsAndBiasesData()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| D | depthwise_conv_plus_1x1_conv.cc | 77 const int s_ch = s * 4 + j; in UploadWeights() local
79 if (s_ch < conv_attr.weights.shape.i && in UploadWeights()
82 conv_attr.weights.shape.LinearIndex({d_ch, 0, 0, s_ch}); in UploadWeights()
|