Searched refs:spatial_remap (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| H A D | weights_conversion.cc | 33 return i_aligned * o_aligned * weight_desc.spatial_remap.size(); in GetTotalElementsCountForLayout()
95 weights, dst_weight_desc.spatial_remap, in RearrangeWeights()
100 weights, dst_weight_desc.spatial_remap, in RearrangeWeights()
108 weights, dst_weight_desc.spatial_remap, in RearrangeWeights()
113 weights, dst_weight_desc.spatial_remap, in RearrangeWeights()
183 weights, dst_weight_desc.spatial_remap, in RearrangeWeights()
188 weights, dst_weight_desc.spatial_remap, in RearrangeWeights()
196 weights, dst_weight_desc.spatial_remap, in RearrangeWeights()
201 weights, dst_weight_desc.spatial_remap, in RearrangeWeights()
|
| H A D | weights_conversion.h | 336 const std::vector<int>& spatial_remap, absl::Span<T> dst) { in RearrangeWeightsToOICustomSpatialI4O4() argument
345 const int kernel_index = spatial_remap[y * weights.shape.w + x]; in RearrangeWeightsToOICustomSpatialI4O4()
372 const std::vector<int>& spatial_remap, absl::Span<T> dst) { in RearrangeWeightsToOICustomSpatialI4O4() argument
383 spatial_remap[(z * weights.shape.h + y) * weights.shape.w + x]; in RearrangeWeightsToOICustomSpatialI4O4()
414 const std::vector<int>& spatial_remap, absl::Span<T> dst) { in RearrangeWeightsToOICustomSpatialO4I4() argument
423 const int kernel_index = spatial_remap[y * weights.shape.w + x]; in RearrangeWeightsToOICustomSpatialO4I4()
450 const std::vector<int>& spatial_remap, absl::Span<T> dst) { in RearrangeWeightsToOICustomSpatialO4I4() argument
461 spatial_remap[(z * weights.shape.h + y) * weights.shape.w + x]; in RearrangeWeightsToOICustomSpatialO4I4()
|
| H A D | weights_layout.cc | 51 IsCustomSpatial() ? spatial_remap == t.spatial_remap : true; in operator ==()
|
| H A D | weights_layout.h | 44 std::vector<int> spatial_remap; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| H A D | conv_weights_converter.cc | 53 std::vector<int32_t> remap(weights_desc_.spatial_remap.size()); in GetConverterToConvWeightsCode()
55 remap[i] = weights_desc_.spatial_remap[i]; in GetConverterToConvWeightsCode()
|
| H A D | convolution_transposed_3x3_thin.h | 52 desc.spatial_remap = GetSpatialWeightsRemap(); in GetWeightsDescription()
|
| H A D | convolution_transposed_4x4.h | 60 desc.spatial_remap = GetSpatialWeightsRemap(); in GetWeightsDescription()
|
| H A D | convolution_transposed_3x3.h | 58 desc.spatial_remap = GetSpatialWeightsRemap(); in GetWeightsDescription()
|
| H A D | conv_weights_converter_test_util.cc | 271 weight_desc.spatial_remap = {10, 11, 14, 15, 8, 9, 12, 13, in ConverterToConvTransposedWeights4x4Test()
|