Searched refs:GetTotalElementsCountForLayout (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | weights_conversion.cc | 20 uint GetTotalElementsCountForLayout(const WeightsDescription& weight_desc, in GetTotalElementsCountForLayout() function
44 GetTotalElementsCountForLayout(dst_weight_desc, weights.shape); in RearrangeWeights()
|
| D | weights_conversion.h | 327 uint GetTotalElementsCountForLayout(const WeightsDescription& weight_desc,
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | convolution_transposed.h | 101 GetTotalElementsCountForLayout(GetWeightsDescription(), weights.shape); in UploadWeights()
|
| D | convolution_transposed_3x3_thin.cc | 206 GetTotalElementsCountForLayout(GetWeightsDescription(), weights.shape); in UploadWeights()
|
| D | convolution_transposed_4x4.cc | 395 GetTotalElementsCountForLayout(GetWeightsDescription(), weights.shape); in UploadWeights()
|
| D | conv_weights_converter_test_util.cc | 52 GetTotalElementsCountForLayout(weight_desc, weights.shape); in ConvolutionWeightsConverterTest()
|
| D | convolution_transposed_3x3.cc | 390 GetTotalElementsCountForLayout(GetWeightsDescription(), weights.shape); in UploadWeights()
|
| D | conv_powervr.h | 257 GetTotalElementsCountForLayout(GetWeightsDescription(), weights.shape); in UploadWeights()
|
| D | conv_metal.cc | 596 GetTotalElementsCountForLayout(weights_desc, weights.shape); in ReorderWeightsForConv()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | operation_selector.cc | 378 GetTotalElementsCountForLayout(weights_desc, weights_shape)), in GPUOperationFromNode()
|