Searched refs:out_group_size (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | weights_conversion.h | 37 const tflite::gpu::Tensor<OHWI, S>& weights, int out_group_size, in RearrangeWeightsToOHWIOGroupI4O4() argument
41 const int dst_groups = DivideRoundUp(dst_slices, out_group_size); in RearrangeWeightsToOHWIOGroupI4O4()
48 for (int d_group = 0; d_group < out_group_size; ++d_group) { in RearrangeWeightsToOHWIOGroupI4O4()
53 const int d_ch = (d * out_group_size + d_group) * 4 + i; in RearrangeWeightsToOHWIOGroupI4O4()
73 const tflite::gpu::Tensor<OHWI, S>& weights, int out_group_size, in RearrangeWeightsToOHWIOGroupO4I4() argument
77 const int dst_groups = DivideRoundUp(dst_slices, out_group_size); in RearrangeWeightsToOHWIOGroupO4I4()
84 for (int d_group = 0; d_group < out_group_size; ++d_group) { in RearrangeWeightsToOHWIOGroupO4I4()
89 const int d_ch = (d * out_group_size + d_group) * 4 + j; in RearrangeWeightsToOHWIOGroupO4I4()
109 const tflite::gpu::Tensor<OHWDI, S>& weights, int out_group_size, in RearrangeWeightsToODHWIOGroupI4O4() argument
113 const int dst_groups = DivideRoundUp(dst_slices, out_group_size); in RearrangeWeightsToODHWIOGroupI4O4()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_weights_converter.cc | 179 const int out_group_size = weights_desc_.GetOutputGroupSize(); in GetGridSize() local
181 DivideRoundUp(AlignByN(src_[0]->Batch(), 4 * out_group_size), 4); in GetGridSize()
|