Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DBidirectionalSequenceLSTM.h153 static constexpr int kBwOutputCellStateTensor = 5; variable
H A DBidirectionalSequenceLSTM.cpp204 bw_output_cell_state_ = GetOutput(operation, operands, kBwOutputCellStateTensor - delta); in BidirectionalSequenceLSTM()
/aosp12/packages/modules/NeuralNetworks/common/
H A DCpuExecutor.cpp1027 operands[outs[BidirectionalSequenceLSTM::kBwOutputCellStateTensor - delta]]; in executeOperation()