| /external/armnn/delegate/classic/src/ |
| 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
|
| D | Gather.hpp | 55 auto inputDimensions = static_cast<int32_t>(inputTensorInfo.GetNumDimensions()); in VisitGatherOperator() local
|
| /external/armnn/src/backends/neon/workloads/ |
| D | NeonStackWorkload.cpp | 20 int CalcAxis(const unsigned int axis, const unsigned int inputDimensions) in CalcAxis()
|
| /external/armnn/src/backends/cl/workloads/ |
| D | ClStackWorkload.cpp | 22 int CalcAxis(const unsigned int axis, const unsigned int inputDimensions) in CalcAxis()
|
| /external/android-nn-driver/test/ |
| 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
|
| 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
|
| /external/android-nn-driver/test/1.3/ |
| 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
|
| /external/armnn/src/backends/aclCommon/ |
| D | ArmComputeTensorUtils.cpp | 83 arm_compute::Coordinates BuildArmComputeReductionCoordinates(size_t inputDimensions, in BuildArmComputeReductionCoordinates()
|
| /external/android-nn-driver/ |
| 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
|
| /external/armnn/src/armnnTfLiteParser/ |
| 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
|
| /external/armnn/shim/sl/canonical/ |
| D | Converter.cpp | 2058 auto inputDimensions = input.GetTensorInfo().GetNumDimensions(); in ConvertGather() local
|