Searched defs:scratchBufferDimensions (Results 1 – 1 of 1) sorted by relevance
106 void LstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in LstmTestImpl()826 hidl_vec<uint32_t> scratchBufferDimensions{batchSize, numUnits * 4}; in LstmNoCifgNoPeepholeNoProjection() local1016 hidl_vec<uint32_t> scratchBufferDimensions{batchSize, numUnits * 3}; in LstmCifgPeepholeNoProjection() local1713 hidl_vec<uint32_t> scratchBufferDimensions{batchSize, numUnits * 4}; in LstmNoCifgPeepholeProjection() local1926 hidl_vec<uint32_t> scratchBufferDimensions{batchSize, numUnits * 3}; in LstmCifgPeepholeNoProjectionBatch2() local2128 hidl_vec<uint32_t> scratchBufferDimensions{batchSize, numUnits * 4}; in LstmNoCifgPeepholeProjectionNoClippingLayerNorm() local2324 hidl_vec<uint32_t> scratchBufferDimensions{batchSize, numUnits * 3}; in LstmCifgPeepholeProjectionNoClippingLayerNorm() local