Searched defs:scratchBufferDimensions (Results 1 – 1 of 1) sorted by relevance
83 void LstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in LstmTestImpl()806 hidl_vec<uint32_t> scratchBufferDimensions{batchSize, numUnits * 4}; in LstmNoCifgNoPeepholeNoProjection() local996 hidl_vec<uint32_t> scratchBufferDimensions{batchSize, numUnits * 3}; in LstmCifgPeepholeNoProjection() local1693 hidl_vec<uint32_t> scratchBufferDimensions{batchSize, numUnits * 4}; in LstmNoCifgPeepholeProjection() local1906 hidl_vec<uint32_t> scratchBufferDimensions{batchSize, numUnits * 3}; in LstmCifgPeepholeNoProjectionBatch2() local2108 hidl_vec<uint32_t> scratchBufferDimensions{batchSize, numUnits * 4}; in LstmNoCifgPeepholeProjectionNoClippingLayerNorm() local2304 hidl_vec<uint32_t> scratchBufferDimensions{batchSize, numUnits * 3}; in LstmCifgPeepholeProjectionNoClippingLayerNorm() local