• Home
  • Raw
  • Download

Lines Matching refs:cellStateInTensorInfo

164     armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, ArmnnType, qScale, qOffset);  in LstmNoCifgNoPeepholeNoProjectionTestImpl()  local
188 tensorHandleFactory.CreateTensorHandle(cellStateInTensorInfo); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
205 AddInputToWorkload(data, info, cellStateInTensorInfo, cellStateInHandle.get()); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
359 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, ArmnnType, qScale, qOffset); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
384 tensorHandleFactory.CreateTensorHandle(cellStateInTensorInfo); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
401 AddInputToWorkload(data, info, cellStateInTensorInfo, cellStateInHandle.get()); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1042 armnn::TensorInfo cellStateInTensorInfo({batchSize, cellSize}, ArmnnType, qScale, qOffset); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1193 tensorHandleFactory.CreateTensorHandle(cellStateInTensorInfo); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1207 AddInputToWorkload(data, info, cellStateInTensorInfo, cellStateInHandle.get()); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1266 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, ArmnnType, qScale, qOffset); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
1291 tensorHandleFactory.CreateTensorHandle(cellStateInTensorInfo); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl()
1308 AddInputToWorkload(data, info, cellStateInTensorInfo, cellStateInHandle.get()); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl()