Searched refs:cellStateOutTensor (Results 1 – 2 of 2) sorted by relevance
| /external/armnn/src/backends/cl/workloads/ |
| D | ClQuantizedLstmWorkload.cpp | 108 …arm_compute::ICLTensor& cellStateOutTensor = static_cast<IClTensorHandle*>(m_Data.m_Outputs… in ClQuantizedLstmWorkload() local
117 &cellStateOutTensor, &outputStateOutTensor); in ClQuantizedLstmWorkload()
|
| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | LstmTestImpl.cpp | 182 auto cellStateOutTensor = MakeTensor<T,2>(cellStateOutTensorInfo, cellStateOutVector); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
386 auto cellStateOutTensor = MakeTensor<T,2>(cellStateOutTensorInfo, cellStateOutVector); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
1214 auto cellStateOutTensor = MakeTensor<T,2>(cellStateOutTensorInfo, cellStateOutVector); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1271 CopyDataToITensorHandle(cellStateOutHandle.get(), &cellStateOutTensor[0][0]); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1329 auto cellStateOutTensor = MakeTensor<float,2>(cellStateOutTensorInfo, cellStateOutVector); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
1614 auto cellStateOutTensor = MakeTensor<int16_t, 2>(cellStateInfo, cellStateOutVector); in QuantizedLstmTestImpl() local
1825 auto cellStateOutTensor = MakeTensor<int16_t, 2>(cellStateInfo, cellStateOutVector); in QLstmTestImpl() local
2065 auto cellStateOutTensor = MakeTensor<int16_t, 2>(cellStateInfo, cellStateOutVector); in QLstmTestImpl1() local
2337 auto cellStateOutTensor = MakeTensor<int16_t, 2>(cellStateInfo, cellStateOutVector); in QLstmTestImpl2() local
|