| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 1041 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in argMinMaxTest() local
1077 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in dequantizeOpTest() local
1100 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in expandDimsTest() local
1123 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in gatherTest() local
1150 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in quantizeOpTest() local
1229 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in splitTest() local
1258 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in tileTest() local
1282 uint32_t inputDimensions[4] = {4, 5, 6, 7}; in topkV2Test() local
1302 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in simpleMathOpTest() local
1392 uint32_t inputDimensions[4] = {2, 2, 2, 2}; in TEST() local
[all …]
|
| /aosp_15_r20/external/android-nn-driver/test/ |
| H A D | Lstm.hpp | 83 void LstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in LstmTestImpl()
387 void QuantizedLstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in QuantizedLstmTestImpl()
672 hidl_vec<uint32_t> inputDimensions{batchSize, inputSize}; in LstmNoCifgNoPeepholeNoProjection() local
868 hidl_vec<uint32_t> inputDimensions{batchSize, inputSize}; in LstmCifgPeepholeNoProjection() local
1058 hidl_vec<uint32_t> inputDimensions{batchSize, inputSize}; in LstmNoCifgPeepholeProjection() local
1778 hidl_vec<uint32_t> inputDimensions{batchSize, inputSize}; in LstmCifgPeepholeNoProjectionBatch2() local
1971 hidl_vec<uint32_t> inputDimensions{batchSize, inputSize}; in LstmNoCifgPeepholeProjectionNoClippingLayerNorm() local
2173 hidl_vec<uint32_t> inputDimensions{batchSize, inputSize}; in LstmCifgPeepholeProjectionNoClippingLayerNorm() local
2369 hidl_vec<uint32_t> inputDimensions{batchSize, inputSize}; in QuantizedLstm() local
|
| H A D | UnidirectionalSequenceLstm.hpp | 84 void UnidirectionalSequenceLstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in UnidirectionalSequenceLstmTestImpl()
374 hidl_vec<uint32_t> inputDimensions{batchSize, timeSize, inputSize}; in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
577 hidl_vec<uint32_t> inputDimensions{timeSize, batchSize, inputSize}; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
782 hidl_vec<uint32_t> inputDimensions{batchSize, timeSize, inputSize}; in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
1022 hidl_vec<uint32_t> inputDimensions{batchSize, timeSize, inputSize}; in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
1237 hidl_vec<uint32_t> inputDimensions{batchSize, timeSize, inputSize}; in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTestImpl() local
|
| /aosp_15_r20/external/android-nn-driver/test/1.3/ |
| H A D | QLstm.cpp | 62 void QLstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in QLstmTestImpl()
529 hidl_vec<uint32_t> inputDimensions{batchSize, inputSize}; in QLstmWithProjection() local
687 hidl_vec<uint32_t> inputDimensions{batchSize, inputSize}; in QLstmWithNoProjection() local
848 hidl_vec<uint32_t> inputDimensions{batchSize, inputSize}; in DynamicOutputQLstmWithNoProjection() local
|
| /aosp_15_r20/external/armnn/delegate/classic/src/ |
| H A D | Split.hpp | 52 auto inputDimensions = static_cast<int32_t>(inputTensorInfo.GetNumDimensions()); in VisitSplitOperator() local
192 auto inputDimensions = static_cast<int32_t>(inputTensorInfo.GetNumDimensions()); in VisitSplitVOperator() local
|
| H A D | Gather.hpp | 55 auto inputDimensions = static_cast<int32_t>(inputTensorInfo.GetNumDimensions()); in VisitGatherOperator() local
|
| /aosp_15_r20/external/armnn/src/backends/cl/workloads/ |
| H A D | ClStackWorkload.cpp | 22 int CalcAxis(const unsigned int axis, const unsigned int inputDimensions) in CalcAxis()
|
| /aosp_15_r20/external/armnn/src/backends/neon/workloads/ |
| H A D | NeonStackWorkload.cpp | 20 int CalcAxis(const unsigned int axis, const unsigned int inputDimensions) in CalcAxis()
|
| /aosp_15_r20/external/armnn/src/backends/aclCommon/ |
| H A D | ArmComputeTensorUtils.cpp | 83 arm_compute::Coordinates BuildArmComputeReductionCoordinates(size_t inputDimensions, in BuildArmComputeReductionCoordinates()
|
| /aosp_15_r20/external/armnn/src/armnnTfLiteParser/ |
| H A D | TfLiteParser.cpp | 3954 auto inputDimensions = static_cast<int32_t>(inputTensorInfo.GetNumDimensions()); in ParseSplit() local
4074 auto inputDimensions = static_cast<int32_t>(inputTensorInfo.GetNumDimensions()); in ParseSplitV() local
4233 auto inputDimensions = static_cast<int32_t>(inputTensorInfo.GetNumDimensions()); in ParseArgMinMax() local
4287 auto inputDimensions = static_cast<int32_t>(inputTensorInfo.GetNumDimensions()); in ParseGather() local
|
| /aosp_15_r20/external/android-nn-driver/ |
| H A D | ConversionUtils_1_2.hpp | 219 auto inputDimensions = static_cast<int32_t>(input.GetTensorInfo().GetNumDimensions()); in ConvertChannelShuffle() local
898 auto inputDimensions = input.GetTensorInfo().GetNumDimensions(); in ConvertGather() local
|
| /aosp_15_r20/external/armnn/shim/sl/canonical/ |
| H A D | Converter.cpp | 2058 auto inputDimensions = input.GetTensorInfo().GetNumDimensions(); in ConvertGather() local
|