Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceLSTM.h81 static constexpr int kBwInputToCellWeightsTensor = 20; variable
DBidirectionalSequenceLSTM.cpp120 bw_input_to_cell_weights_ = GetInput(operation, operands, kBwInputToCellWeightsTensor); in BidirectionalSequenceLSTM()
226 kBwInputToCellWeightsTensor, in Prepare()