Searched refs:cellStateOut (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 411 float* cellStateOut; in execute() local
414 cellStateOut = context->getOutputBuffer<float>(kCellStateOutTensor); in execute()
419 cellStateOut = cellStateOutBuffer.data(); in execute()
455 cellStateOut, context->getOutputBuffer<float>(kOutputTensor), in execute()
463 _Float16* cellStateOut; in execute() local
466 cellStateOut = context->getOutputBuffer<_Float16>(kCellStateOutTensor); in execute()
471 cellStateOut = cellStateOutBuffer.data(); in execute()
508 outputStateOut, cellStateOut, context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1099 RunTimeOperandInfo& cellStateOut = operands[outs[LSTMCell::kCellStateOutTensor]]; in executeOperation() local
1109 setInfoAndAllocateIfNeeded(&cellStateOut, cellStateShape, &result) && in executeOperation()
1673 RunTimeOperandInfo& cellStateOut = in executeOperation() local
1682 setInfoAndAllocateIfNeeded(&cellStateOut, cellStateOutShape, &result) && in executeOperation()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 2646 ANeuralNetworksOperandType cellStateOut = floatTensor2D; in TEST() local
2673 {scratch, outputStateOut, cellStateOut, output}); in TEST()
2734 ANeuralNetworksOperandType cellStateOut = floatTensor2D; in lstmTestV1_2() local
2765 {scratch, outputStateOut, cellStateOut, output}); in lstmTestV1_2()
|