Searched refs:kFwRecurrentToOutputWeightsTensor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | bidirectional_sequence_lstm.cc | 52 constexpr int kFwRecurrentToOutputWeightsTensor = 8; variable
393 kFwRecurrentToCellWeightsTensor, kFwRecurrentToOutputWeightsTensor, in CheckInputTensorDimensions()
463 context, GetInputSafe(context, node, kFwRecurrentToOutputWeightsTensor, in Prepare()
1009 context, GetInputSafe(context, node, kFwRecurrentToOutputWeightsTensor, in Eval()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | BidirectionalSequenceLSTM.h | 56 static constexpr int kFwRecurrentToOutputWeightsTensor = 8; variable
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BidirectionalSequenceLSTM.cpp | 96 GetInput(operation, operands, kFwRecurrentToOutputWeightsTensor); in BidirectionalSequenceLSTM()
220 kFwRecurrentToOutputWeightsTensor, in Prepare()
|