Home
last modified time | relevance | path

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

/aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/
DBidirectionalSequenceLSTM.h54 static constexpr int kFwRecurrentToForgetWeightsTensor = 6; variable
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DBidirectionalSequenceLSTM.cpp93 GetInput(operation, operands, kFwRecurrentToForgetWeightsTensor); in BidirectionalSequenceLSTM()
218 kFwRecurrentToForgetWeightsTensor, in Prepare()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dbidirectional_sequence_lstm.cc50 constexpr int kFwRecurrentToForgetWeightsTensor = 6; variable
392 kFwRecurrentToInputWeightsTensor, kFwRecurrentToForgetWeightsTensor, in CheckInputTensorDimensions()
1001 context, GetInputSafe(context, node, kFwRecurrentToForgetWeightsTensor, in Eval()