| /aosp_15_r20/external/armnn/delegate/test/ |
| H A D | UnidirectionalSequenceLstmTest.cpp | 80 std::vector<float> cellBias = {0., 0., 0., 0.}; in UnidirectionalSequenceLstmTest() local
138 cellBias, in UnidirectionalSequenceLstmTest()
229 std::vector<float> cellBias = {0., 0., 0., 0.}; in UnidirectionalSequenceLstmTimeMajorTest() local
287 cellBias, in UnidirectionalSequenceLstmTimeMajorTest()
408 std::vector<float> cellBias = { -0.024379363f, 0.0055531194f, 0.23377132f, in UnidirectionalSequenceLstmNoCifgWithPeepholeWithProjectionTest() local
475 cellBias, in UnidirectionalSequenceLstmNoCifgWithPeepholeWithProjectionTest()
554 std::vector<float> cellBias = {0., 0., 0., 0.}; in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest() local
612 cellBias, in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest()
705 std::vector<float> cellBias = { -0.05f, 0.72f, 0.25f, 0.08f, 0.1f }; in UnidirectionalSequenceLstmNoCifgWithPeepholeWithProjectionWithLayerNormTest() local
767 cellBias, in UnidirectionalSequenceLstmNoCifgWithPeepholeWithProjectionWithLayerNormTest()
[all …]
|
| H A D | LstmTestHelper.hpp | 49 const std::vector<T>& cellBias, in CreateLstmTfLiteModel()
326 … flatBufferBuilder.CreateVector(reinterpret_cast<const uint8_t *>(cellBias.data()), in CreateLstmTfLiteModel()
327 sizeof(T) * cellBias.size()))); in CreateLstmTfLiteModel()
573 const std::vector<T>& cellBias, in LstmTestImpl()
619 cellBias, in LstmTestImpl()
|
| H A D | UnidirectionalSequenceLstmTestHelper.hpp | 60 const std::vector<float>& cellBias, in CreateUnidirectionalSequenceLstmTfLiteModel()
355 … flatBufferBuilder.CreateVector(reinterpret_cast<const uint8_t*>(cellBias.data()), in CreateUnidirectionalSequenceLstmTfLiteModel()
356 sizeof(float) * cellBias.size()))); in CreateUnidirectionalSequenceLstmTfLiteModel()
604 const std::vector<float>& cellBias, in UnidirectionalSequenceLstmTestImpl()
653 cellBias, in UnidirectionalSequenceLstmTestImpl()
|
| H A D | LstmTest.cpp | 96 std::vector<float> cellBias {0., 0., 0., 0.}; in LstmTest() local
146 cellBias, in LstmTest()
|
| /aosp_15_r20/external/armnn/src/armnn/test/ |
| H A D | ConstTensorLayerVisitor.cpp | 635 ConstTensor cellBias( variable
653 params.m_CellBias = &cellBias;
717 ConstTensor cellBias( variable
735 params.m_CellBias = &cellBias;
798 ConstTensor cellBias( variable
834 params.m_CellBias = &cellBias;
902 ConstTensor cellBias( variable
938 params.m_CellBias = &cellBias;
1007 ConstTensor cellBias( variable
1037 params.m_CellBias = &cellBias;
[all …]
|
| /aosp_15_r20/external/armnn/src/armnnSerializer/test/ |
| H A D | LstmSerializationTests.cpp | 224 armnn::ConstTensor cellBias(inputWeightsInfo3, cellBiasData);
237 params.m_CellBias = &cellBias;
328 armnn::ConstTensor cellBias(tensorInfo20, cellBiasData);
371 params.m_CellBias = &cellBias;
475 armnn::ConstTensor cellBias(tensorInfo20, cellBiasData);
530 params.m_CellBias = &cellBias;
1260 armnn::ConstTensor cellBias(tensorInfo20, cellBiasData);
1303 params.m_CellBias = &cellBias;
1551 armnn::ConstTensor cellBias(cellBiasInfo, cellBiasData);
1573 params.m_CellBias = &cellBias;
[all …]
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
| H A D | UnidirectionalSequenceLstmTestImpl.cpp | 136 std::vector<float> cellBias = {0., 0., 0., 0.}; in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local
165 AllocateAndCopyDataToITensorHandle(&cellBiasTensor, cellBias.data()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
326 std::vector<float> cellBias = { 0., 0., 0., 0. }; in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
353 AllocateAndCopyDataToITensorHandle(&cellBiasTensor, cellBias.data()); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
515 std::vector<float> cellBias = { 0., 0., 0., 0. }; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
542 AllocateAndCopyDataToITensorHandle(&cellBiasTensor, cellBias.data()); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
772 std::vector<float> cellBias = { -0.024379363f, 0.0055531194f, 0.23377132f, in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local
863 AllocateAndCopyDataToITensorHandle(&cellBiasTensor, cellBias.data()); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest()
1016 std::vector<float> cellBias{ -0.05f, 0.72f, 0.25f, 0.08f, 0.1f }; in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest() local
1094 AllocateAndCopyDataToITensorHandle(&cellBiasTensor, cellBias.data()); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest()
[all …]
|
| H A D | LstmTestImpl.cpp | 266 std::vector<float> cellBias = {0., 0., 0., 0.}; in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
295 AllocateAndCopyDataToITensorHandle(&cellBiasTensor, cellBias.data()); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
514 std::vector<float> cellBias = { -0.024379363f, 0.0055531194f, 0.23377132f, 0.033463873f, in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
959 AllocateAndCopyDataToITensorHandle(&cellBiasTensor, cellBias.data()); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1082 std::vector<float> cellBias = {0.f, 0.f, 0.f, 0.f}; in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1129 AllocateAndCopyDataToITensorHandle(&cellBiasTensor, cellBias.data()); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1345 std::vector<float> cellBias = {-0.05f, 0.72f, 0.25f, 0.08f}; //{numUnits} in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
1424 AllocateAndCopyDataToITensorHandle(&cellBiasTensor, cellBias.data()); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl()
1604 std::vector<int32_t> cellBias = {39481, 48624, 48976, -21419}; in QuantizedLstmTestImpl() local
1636 AllocateAndCopyDataToITensorHandle(&cellBiasTensor, cellBias.data()); in QuantizedLstmTestImpl()
[all …]
|
| /aosp_15_r20/external/armnn/src/armnnSerializer/ |
| H A D | Serializer.cpp | 726 auto cellBias = CreateConstTensorInfo(constants[i++]); //CellBias in SerializeLstmLayer() local
788 cellBias, in SerializeLstmLayer()
1532 auto cellBias = CreateConstTensorInfo(constants[i++]); //CellBias in SerializeQLstmLayer() local
1598 cellBias, in SerializeQLstmLayer()
1646 auto cellBias = CreateConstTensorInfo(constants[i++]); in SerializeQuantizedLstmLayer() local
1661 cellBias, in SerializeQuantizedLstmLayer()
1705 auto cellBias = CreateConstTensorInfo(constants[i++]); //CellBias in SerializeUnidirectionalSequenceLstmLayer() local
1765 cellBias, in SerializeUnidirectionalSequenceLstmLayer()
|
| H A D | ArmnnSchema.fbs | 806 cellBias:ConstTensor;
853 cellBias:ConstTensor;
914 cellBias:ConstTensor;
|
| /aosp_15_r20/external/armnn/src/armnnDeserializer/ |
| H A D | Deserializer.cpp | 3250 armnn::ConstTensor cellBias = ToConstTensor(flatBufferInputParams->cellBias()); in ParseLstm() local
3260 lstmInputParams.m_CellBias = &cellBias; in ParseLstm()
3389 armnn::ConstTensor cellBias = ToConstTensor(flatBufferInputParams->cellBias()); in ParseQLstm() local
3399 qLstmInputParams.m_CellBias = &cellBias; in ParseQLstm()
3515 armnn::ConstTensor cellBias = ToConstTensor(flatBufferInputParams->cellBias()); in ParseQuantizedLstm() local
3528 lstmInputParams.m_CellBias = &cellBias; in ParseQuantizedLstm()
3815 armnn::ConstTensor cellBias = ToConstTensor(flatBufferInputParams->cellBias()); in ParseUnidirectionalSequenceLstm() local
3825 lstmInputParams.m_CellBias = &cellBias; in ParseUnidirectionalSequenceLstm()
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/ |
| H A D | WorkloadFactory.cpp | 650 const TensorInfo& cellBias in IsLayerConfigurationSupported() local
664 paramsInfo.m_CellBias = &cellBias; in IsLayerConfigurationSupported()
1429 const TensorInfo& cellBias in IsLayerConfigurationSupported() local
1443 paramsInfo.m_CellBias = &cellBias; in IsLayerConfigurationSupported()
|
| /aosp_15_r20/external/armnn/generated/ |
| H A D | ArmnnSchema_generated.h | 8176 const armnnSerializer::ConstTensor *cellBias() const {
8235 verifier.VerifyTable(cellBias()) &&
8291 void add_cellBias(flatbuffers::Offset<armnnSerializer::ConstTensor> cellBias) {
8292 fbb_.AddOffset(LstmInputParams::VT_CELLBIAS, cellBias);
8353 flatbuffers::Offset<armnnSerializer::ConstTensor> cellBias = 0,
8381 builder_.add_cellBias(cellBias);
8603 const armnnSerializer::ConstTensor *cellBias() const {
8662 verifier.VerifyTable(cellBias()) &&
8718 void add_cellBias(flatbuffers::Offset<armnnSerializer::ConstTensor> cellBias) {
8719 fbb_.AddOffset(QLstmInputParams::VT_CELLBIAS, cellBias);
[all …]
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 2672 ANeuralNetworksOperandType cellBias = floatTensor1D; in TEST() local
2702 cellBias, in TEST()
2756 ANeuralNetworksOperandType cellBias = floatTensor1D; in lstmTestV1_2() local
2790 cellBias, in lstmTestV1_2()
|