Home
last modified time | relevance | path

Searched refs:kFwOutputCellStateTensor (Results 1 – 3 of 3) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DBidirectionalSequenceLSTM.h151 static constexpr int kFwOutputCellStateTensor = 3; variable
H A DBidirectionalSequenceLSTM.cpp201 fw_output_cell_state_ = GetOutput(operation, operands, kFwOutputCellStateTensor - delta); in BidirectionalSequenceLSTM()
/aosp12/packages/modules/NeuralNetworks/common/
H A DCpuExecutor.cpp1022 operands[outs[BidirectionalSequenceLSTM::kFwOutputCellStateTensor - delta]]; in executeOperation()