Searched refs:ValidateTensorNumDimNumElem (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/ |
| H A D | WorkloadData.cpp | 461 void QueueDescriptor::ValidateTensorNumDimNumElem(const TensorInfo& tensorInfo, in ValidateTensorNumDimNumElem() function in armnn::QueueDescriptor
2019 ValidateTensorNumDimNumElem(workloadInfo.m_InputTensorInfos[0], 2, (n_batch * n_input), in Validate()
2022 ValidateTensorNumDimNumElem(workloadInfo.m_InputTensorInfos[1], 2, (n_batch * n_output), in Validate()
2025 ValidateTensorNumDimNumElem(workloadInfo.m_InputTensorInfos[2], 2, (n_batch * n_cell), in Validate()
2029 … ValidateTensorNumDimNumElem(workloadInfo.m_OutputTensorInfos[0], 2, (n_batch * scratchBufferSize), in Validate()
2032 ValidateTensorNumDimNumElem(workloadInfo.m_OutputTensorInfos[1], 2, (n_batch * n_output), in Validate()
2035 ValidateTensorNumDimNumElem(workloadInfo.m_OutputTensorInfos[2], 2, (n_batch * n_cell), in Validate()
2038 ValidateTensorNumDimNumElem(workloadInfo.m_OutputTensorInfos[3], 2, (n_batch * n_output), in Validate()
2044 ValidateTensorNumDimNumElem(m_InputToInputWeights->GetTensorInfo(), 2, in Validate()
2049 ValidateTensorNumDimNumElem(m_InputToForgetWeights->GetTensorInfo(), 2, in Validate()
[all …]
|
| /aosp_15_r20/external/armnn/include/armnn/backends/ |
| H A D | WorkloadData.hpp | 37 void ValidateTensorNumDimNumElem(const TensorInfo& tensorInfo,
|