Home
last modified time | relevance | path

Searched refs:kFwInputActivationStateTensor (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/
DBidirectionalSequenceLSTM.h105 static constexpr int kFwInputActivationStateTensor = 35; variable
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DBidirectionalSequenceLSTM.cpp113 fw_activation_state_ = GetInput(operation, operands, kFwInputActivationStateTensor); in BidirectionalSequenceLSTM()
233 kFwInputActivationStateTensor, in Prepare()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dbidirectional_sequence_lstm.cc101 constexpr int kFwInputActivationStateTensor = 35; variable
536 GetVariableInput(context, node, kFwInputActivationStateTensor); in Prepare()
1039 GetVariableInput(context, node, kFwInputActivationStateTensor); in Eval()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dpropagate_fixed_sizes.cc1208 constexpr int kFwInputActivationStateTensor = 35; in ProcessBidirectionalSequenceLstmOperator() local
1213 model->GetArray(op->inputs[kFwInputActivationStateTensor]).buffer.reset(); in ProcessBidirectionalSequenceLstmOperator()