Home
last modified time | relevance | path

Searched refs:cellStateInVector (Results 1 – 4 of 4) sorted by relevance

/external/armnn/src/backends/backendsCommon/test/layerTests/
DUnidirectionalSequenceLstmTestImpl.cpp49 std::vector<T> cellStateInVector(batchSize * numUnits, T()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local
202 CopyDataToITensorHandle(cellStateInHandle.get(), cellStateInVector.data()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
245 std::vector<T> cellStateInVector(batchSize * numUnits, T()); in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
390 CopyDataToITensorHandle(cellStateInHandle.get(), cellStateInVector.data()); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
434 std::vector<T> cellStateInVector(batchSize * numUnits, T()); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
579 CopyDataToITensorHandle(cellStateInHandle.get(), cellStateInVector.data()); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
695 std::vector<float> cellStateInVector(batchSize * numUnits, 0.f); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local
910 CopyDataToITensorHandle(cellStateInHandle.get(), cellStateInVector.data()); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest()
947 std::vector<float> cellStateInVector(batchSize * numUnits, 0.f); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest() local
1150 CopyDataToITensorHandle(cellStateInHandle.get(), cellStateInVector.data()); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest()
[all …]
DLstmTestImpl.cpp175 std::vector<T> cellStateInVector(batchSize * numUnits, T()); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
329 CopyDataToITensorHandle(cellStateInHandle.get(), cellStateInVector.data()); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
371 std::vector<T> cellStateInVector(batchSize * numUnits, T()); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
1002 CopyDataToITensorHandle(cellStateInHandle.get(), cellStateInVector.data()); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1056 std::vector<float> cellStateInVector(batchSize * cellSize, 0.f); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1227 CopyDataToITensorHandle(cellStateInHandle.get(), cellStateInVector.data()); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1278 std::vector<float> cellStateInVector(batchSize * numUnits, 0.f); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
1479 CopyDataToITensorHandle(cellStateInHandle.get(), cellStateInVector.data()); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl()
1539 std::vector<int16_t> cellStateInVector = {876, 1034, 955, -909, 761, 1029, 796, -1036}; // 13 in QuantizedLstmTestImpl() local
1668 CopyDataToITensorHandle(cellStateInHandle.get(), cellStateInVector.data()); in QuantizedLstmTestImpl()
[all …]
/external/armnn/src/backends/backendsCommon/test/
DQuantizedLstmEndToEndTestImpl.cpp186 std::vector<int16_t> cellStateInVector = {876, 1034, 955, -909, 761, 1029, 796, -1036}; in QuantizedLstmEndToEnd() local
223 inputTensors.push_back({1, ConstTensor(inputTensorInfo1, cellStateInVector.data())}); in QuantizedLstmEndToEnd()
DQLstmEndToEndTestImpl.cpp204 const std::vector<int16_t> cellStateInVector = {0, 0, 0, 0, 0, 0, 0, 0}; in QLstmEndToEnd() local
250 …inputTensors.push_back({2, ConstTensor(runtime->GetInputTensorInfo(netId, 2), cellStateInVector.da… in QLstmEndToEnd()