Home
last modified time | relevance | path

Searched refs:spatial_remap (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dweights_conversion.cc33 return i_aligned * o_aligned * weight_desc.spatial_remap.size(); in GetTotalElementsCountForLayout()
81 weights, dst_weight_desc.spatial_remap, in RearrangeWeights()
86 weights, dst_weight_desc.spatial_remap, in RearrangeWeights()
94 weights, dst_weight_desc.spatial_remap, in RearrangeWeights()
99 weights, dst_weight_desc.spatial_remap, in RearrangeWeights()
170 weights, dst_weight_desc.spatial_remap, in RearrangeWeights()
175 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()
Dweights_conversion.h336 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()
Dweights_layout.h41 std::vector<int> spatial_remap; member
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_weights_converter.cc59 std::vector<int32_t> remap(conv_weights_desc.spatial_remap.size()); in GetConverterToConvWeightsCode()
61 remap[i] = conv_weights_desc.spatial_remap[i]; in GetConverterToConvWeightsCode()
Dconvolution_transposed_3x3_thin.h54 desc.spatial_remap = GetSpatialWeightsRemap(); in GetWeightsDescription()
Dconvolution_transposed_4x4.h59 desc.spatial_remap = GetSpatialWeightsRemap(); in GetWeightsDescription()
Dconvolution_transposed_3x3.h57 desc.spatial_remap = GetSpatialWeightsRemap(); in GetWeightsDescription()
Dconv_weights_converter_test_util.cc246 weight_desc.spatial_remap = {10, 11, 14, 15, 8, 9, 12, 13, in ConverterToConvTransposedWeights4x4Test()