Home
last modified time | relevance | path

Searched refs:cellToInputWeightsTensor (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefLstmWorkload.cpp143 std::unique_ptr<Decoder<float>> cellToInputWeightsTensor; in Execute() local
193 cellToInputWeightsTensor = MakeDecoder<float>( in Execute()
229 cellToInputWeightsTensor, in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.cpp174 std::unique_ptr<Decoder<float>> cellToInputWeightsTensor; in Execute() local
221 cellToInputWeightsTensor = MakeDecoder<float>( in Execute()
262 cellToInputWeightsTensor, in Execute()
H A DLstm.cpp34 std::unique_ptr<Decoder<float>>& cellToInputWeightsTensor, in LstmImpl() argument
131 VectorBatchVectorCwiseProductAccumulate(*cellToInputWeightsTensor, in LstmImpl()
H A DLstm.hpp38 std::unique_ptr<Decoder<float>>& cellToInputWeightsTensor,
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DUnidirectionalSequenceLstmTestImpl.cpp148 armnn::ScopedTensorHandle cellToInputWeightsTensor(tensorInfo4); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local
162 AllocateAndCopyDataToITensorHandle(&cellToInputWeightsTensor, cellToInputWeights.data()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
842 armnn::ScopedTensorHandle cellToInputWeightsTensor(tensorInfo6); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local
860 AllocateAndCopyDataToITensorHandle(&cellToInputWeightsTensor, cellToInputWeights.data()); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest()
878 data.m_CellToInputWeights = &cellToInputWeightsTensor; in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest()
1068 armnn::ScopedTensorHandle cellToInputWeightsTensor(tensorInfo5); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest() local
1091 AllocateAndCopyDataToITensorHandle(&cellToInputWeightsTensor, cellToInputWeights.data()); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest()
1114 data.m_CellToInputWeights = &cellToInputWeightsTensor; in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest()
1745 armnn::ScopedTensorHandle cellToInputWeightsTensor(tensorInfoNum); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest() local
1763 AllocateAndCopyDataToITensorHandle(&cellToInputWeightsTensor, cellToInputWeights.data()); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest()
[all …]
H A DLstmTestImpl.cpp278 armnn::ScopedTensorHandle cellToInputWeightsTensor(tensorInfo4); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
292 AllocateAndCopyDataToITensorHandle(&cellToInputWeightsTensor, cellToInputWeights.data()); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
938 armnn::ScopedTensorHandle cellToInputWeightsTensor(tensorInfo20); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
956 AllocateAndCopyDataToITensorHandle(&cellToInputWeightsTensor, cellToInputWeights.data()); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
974 data.m_CellToInputWeights = &cellToInputWeightsTensor; in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1398 armnn::ScopedTensorHandle cellToInputWeightsTensor(tensorInfo4); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
1421 AllocateAndCopyDataToITensorHandle(&cellToInputWeightsTensor, cellToInputWeights.data()); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl()
1444 data.m_CellToInputWeights = &cellToInputWeightsTensor; in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DWorkloadDataValidation.cpp541 armnn::ScopedTensorHandle cellToInputWeightsTensor(tensorInfo4);
563 data.m_CellToInputWeights = &cellToInputWeightsTensor;