Searched refs:outputStateInTensor (Results 1 – 2 of 2) sorted by relevance
| /external/armnn/src/backends/cl/workloads/ |
| D | ClQuantizedLstmWorkload.cpp | 106 …const arm_compute::ICLTensor& outputStateInTensor = static_cast<IClTensorHandle*>(m_Data.m_Inputs[… in ClQuantizedLstmWorkload() local
116 … m_OutputGateBiasTensor.get(), &cellStateInTensor, &outputStateInTensor, in ClQuantizedLstmWorkload()
|
| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | LstmTestImpl.cpp | 173 auto outputStateInTensor = MakeTensor<T,2>(outputStateInTensorInfo, outputStateInVector); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
330 CopyDataToITensorHandle(outputStateInHandle.get(), &outputStateInTensor[0][0]); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
377 auto outputStateInTensor = MakeTensor<T,2>(outputStateInTensorInfo, outputStateInVector); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
1046 CopyDataToITensorHandle(outputStateInHandle.get(), &outputStateInTensor[0][0]); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1097 auto outputStateInTensor = MakeTensor<float, 2>(outputStateInTensorInfo, outputStateInVector); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1266 CopyDataToITensorHandle(outputStateInHandle.get(), &outputStateInTensor[0][0]); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1320 auto outputStateInTensor = MakeTensor<float,2>(outputStateInTensorInfo, outputStateInVector); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
1546 CopyDataToITensorHandle(outputStateInHandle.get(), &outputStateInTensor[0][0]); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl()
1610 auto outputStateInTensor = MakeTensor<uint8_t, 2>(outputStateInfo, outputStateInVector); in QuantizedLstmTestImpl() local
1734 CopyDataToITensorHandle(outputStateInHandle.get(), &outputStateInTensor[0][0]); in QuantizedLstmTestImpl()
[all …]
|