Searched refs:spatial_remap (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | weights_conversion.cc | 33 return i_aligned * o_aligned * weight_desc.spatial_remap.size(); in GetTotalElementsCountForLayout()
75 weights, dst_weight_desc.spatial_remap, in RearrangeWeights()
80 weights, dst_weight_desc.spatial_remap, in RearrangeWeights()
88 weights, dst_weight_desc.spatial_remap, in RearrangeWeights()
93 weights, dst_weight_desc.spatial_remap, in RearrangeWeights()
|
| D | weights_conversion.h | 258 const std::vector<int>& spatial_remap, absl::Span<T> dst) { in RearrangeWeightsToOICustomSpatialI4O4() argument
267 const int kernel_index = spatial_remap[y * weights.shape.w + x]; in RearrangeWeightsToOICustomSpatialI4O4()
294 const std::vector<int>& spatial_remap, absl::Span<T> dst) { in RearrangeWeightsToOICustomSpatialO4I4() argument
303 const int kernel_index = spatial_remap[y * weights.shape.w + x]; in RearrangeWeightsToOICustomSpatialO4I4()
|
| D | weights_layout.h | 39 std::vector<int> spatial_remap; member
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_weights_converter.cc | 57 std::vector<int32_t> remap(conv_weights_desc.spatial_remap.size()); in GetConverterToConvWeightsCode()
59 remap[i] = conv_weights_desc.spatial_remap[i]; in GetConverterToConvWeightsCode()
|
| D | convolution_transposed_3x3_thin.h | 54 desc.spatial_remap = GetSpatialWeightsRemap(); in GetWeightsDescription()
|
| D | convolution_transposed_3x3.h | 57 desc.spatial_remap = GetSpatialWeightsRemap(); in GetWeightsDescription()
|
| D | convolution_transposed_4x4.h | 59 desc.spatial_remap = GetSpatialWeightsRemap(); in GetWeightsDescription()
|
| D | conv_weights_converter_test_util.cc | 246 weight_desc.spatial_remap = {10, 11, 14, 15, 8, 9, 12, 13, in ConverterToConvTransposedWeights4x4Test()
|