Searched defs:outputDimensions (Results 1 – 5 of 5) sorted by relevance
| /external/armnn/src/armnn/layers/ |
| D | PreluLayer.cpp | 54 unsigned int outputDimensions = std::max(inputShapeDimensions, alphaShapeDimensions); in InferOutputShapes() local
|
| /external/android-nn-driver/test/ |
| D | Lstm.hpp | 106 void LstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in LstmTestImpl()
408 void QuantizedLstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in QuantizedLstmTestImpl()
836 hidl_vec<uint32_t> outputDimensions{batchSize, outputSize}; in LstmNoCifgNoPeepholeNoProjection() local
1026 hidl_vec<uint32_t> outputDimensions{batchSize, outputSize}; in LstmCifgPeepholeNoProjection() local
1739 hidl_vec<uint32_t> outputDimensions{batchSize, outputSize}; in LstmNoCifgPeepholeProjection() local
1938 hidl_vec<uint32_t> outputDimensions{batchSize, outputSize}; in LstmCifgPeepholeNoProjectionBatch2() local
2140 hidl_vec<uint32_t> outputDimensions{batchSize, outputSize}; in LstmNoCifgPeepholeProjectionNoClippingLayerNorm() local
2336 hidl_vec<uint32_t> outputDimensions{batchSize, outputSize}; in LstmCifgPeepholeProjectionNoClippingLayerNorm() local
2475 hidl_vec<uint32_t> outputDimensions{batchSize, outputSize}; in QuantizedLstm() local
|
| /external/android-nn-driver/test/1.3/ |
| D | QLstm.cpp | 95 void QLstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in QLstmTestImpl()
666 hidl_vec<uint32_t> outputDimensions{batchSize, outputSize}; in QLstmWithProjection() local
826 hidl_vec<uint32_t> outputDimensions{batchSize, outputSize}; in QLstmWithNoProjection() local
987 hidl_vec<uint32_t> outputDimensions{batchSize, outputSize}; in DynamicOutputQLstmWithNoProjection() local
|
| /external/android-nn-driver/ |
| D | ConversionUtils_1_2.hpp | 727 auto outputDimensions = outputInfo.GetNumDimensions(); in ConvertGather() local
|
| /external/armnn/src/armnnTfParser/ |
| D | TfParser.cpp | 2028 std::vector<unsigned int> outputDimensions; in ParseStack() local
|