Searched refs:RearrangeWeights (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | weights_conversion.cc | 39 void RearrangeWeights( in RearrangeWeights() function
|
| D | weights_conversion.h | 330 void RearrangeWeights(
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | convolution_transposed.h | 107 RearrangeWeights(weights, GetWeightsDescription(), weights_type, in UploadWeights()
|
| D | convolution_transposed_3x3_thin.cc | 219 RearrangeWeights(weights, GetWeightsDescription(), weights_type, in UploadWeights()
|
| D | convolution_transposed_4x4.cc | 412 RearrangeWeights(weights, GetWeightsDescription(), weights_type, in UploadWeights()
|
| D | conv_weights_converter_test_util.cc | 56 RearrangeWeights(weights, weight_desc, weights_type, in ConvolutionWeightsConverterTest()
|
| D | convolution_transposed_3x3.cc | 407 RearrangeWeights(weights, GetWeightsDescription(), weights_type, in UploadWeights()
|
| D | conv_powervr.h | 261 RearrangeWeights(weights, GetWeightsDescription(), weights_type, in UploadWeights()
|
| D | conv_metal.cc | 598 RearrangeWeights(weights, weights_desc, weights_type, absl::MakeSpan(result)); in ReorderWeightsForConv()
|