Searched refs:ValidateTensorNumDimensions (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/ |
| H A D | WorkloadData.cpp | 423 void QueueDescriptor::ValidateTensorNumDimensions(const TensorInfo& tensor, in ValidateTensorNumDimensions() function in armnn::QueueDescriptor
467 ValidateTensorNumDimensions(tensorInfo, functionName, numDimension, tensorName); in ValidateTensorNumDimNumElem()
1046 ValidateTensorNumDimensions(inputTensorInfo, descriptorName, 1, "input"); in Validate()
1075 ValidateTensorNumDimensions(outputTensorInfo, descriptorName, 2, "output"); in Validate()
1083 ValidateTensorNumDimensions(weightTensorInfo, descriptorName, 2, "weight"); in Validate()
1091 ValidateTensorNumDimensions(biasTensorInfo, descriptorName, 1, "bias"); in Validate()
1260 ValidateTensorNumDimensions(mean, descriptorName, 1, "mean"); in Validate()
1261 ValidateTensorNumDimensions(variance, descriptorName, 1, "variance"); in Validate()
1262 ValidateTensorNumDimensions(beta, descriptorName, 1, "beta"); in Validate()
1263 ValidateTensorNumDimensions(gamma, descriptorName, 1, "gamma"); in Validate()
[all …]
|
| /aosp_15_r20/external/armnn/include/armnn/backends/ |
| H A D | WorkloadData.hpp | 32 void ValidateTensorNumDimensions(const TensorInfo& tensor,
|