Searched refs:kFwInputToOutputWeightsTensor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | bidirectional_sequence_lstm.cc | 46 constexpr int kFwInputToOutputWeightsTensor = 4; variable
391 kFwInputToCellWeightsTensor, kFwInputToOutputWeightsTensor, in CheckInputTensorDimensions()
443 GetInputSafe(context, node, kFwInputToOutputWeightsTensor, in Prepare()
994 GetInputSafe(context, node, kFwInputToOutputWeightsTensor, in Eval()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | BidirectionalSequenceLSTM.h | 50 static constexpr int kFwInputToOutputWeightsTensor = 4; variable
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BidirectionalSequenceLSTM.cpp | 88 fw_input_to_output_weights_ = GetInput(operation, operands, kFwInputToOutputWeightsTensor); in BidirectionalSequenceLSTM()
217 kFwInputToOutputWeightsTensor, in Prepare()
|