Home
last modified time | relevance | path

Searched refs:kFwRecurrentToCellWeightsTensor (Results 1 – 2 of 2) sorted by relevance

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceLSTM.h58 static constexpr int kFwRecurrentToCellWeightsTensor = 7; variable
DBidirectionalSequenceLSTM.cpp95 fw_recurrent_to_cell_weights_ = GetInput(operation, operands, kFwRecurrentToCellWeightsTensor); in BidirectionalSequenceLSTM()
220 kFwRecurrentToCellWeightsTensor, in Prepare()