Home
last modified time | relevance | path

Searched refs:d_ch (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dweights_conversion.h53 const int d_ch = (d * out_group_size + d_group) * 4 + i; 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()
89 const int d_ch = (d * out_group_size + d_group) * 4 + j; 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()
126 const int d_ch = (d * out_group_size + d_group) * 4 + i; 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()
163 const int d_ch = (d * out_group_size + d_group) * 4 + i; in RearrangeWeightsToI4HWIOOGroupO4() local
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Ddepthwise_conv.h50 const int d_ch = d * 4 + i; in RearrangeWeightsForDWConv2D() local
51 if (d_ch < dst_channels) { in RearrangeWeightsForDWConv2D()
53 {d_ch % weights.shape.o, y, x, d_ch / weights.shape.o}); in RearrangeWeightsForDWConv2D()
123 const int d_ch = d * 4 + i; in RearrangeWeightsForDWConv3D() local
124 if (d_ch < dst_channels) { in RearrangeWeightsForDWConv3D()
126 {d_ch % weights.shape.o, y, x, z, d_ch / weights.shape.o}); in RearrangeWeightsForDWConv3D()
Dconv_constants.h51 const int d_ch = d * 4 + i; 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()
88 const int d_ch = d * 4 + j; 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()
Dconvolution_transposed_thin.h121 const int d_ch = j; 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()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Ddepthwise_conv_plus_1x1_conv.cc52 const int d_ch = d * 4 + i; in UploadWeights() local
53 if (d_ch < dw_attr.weights.shape.i) { in UploadWeights()
55 dw_attr.weights.shape.LinearIndex({0, y, x, d_ch}); in UploadWeights()
78 const int d_ch = d * 4 + i; in UploadWeights() local
80 d_ch < conv_attr.weights.shape.o) { in UploadWeights()
82 conv_attr.weights.shape.LinearIndex({d_ch, 0, 0, s_ch}); in UploadWeights()