Searched refs:kAuxInputTensor (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BidirectionalSequenceRNN.cpp | 67 const bool hasAuxInput = !context->isOmittedInput(kAuxInputTensor); in getLinkingMode()
116 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()
120 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()
125 Shape auxInputShape = context->getInputShape(kAuxInputTensor); in executeTyped()
318 Shape auxInput = context->getInputShape(kAuxInputTensor); in prepare()
|
| D | BidirectionalSequenceLSTM.cpp | 148 aux_input_ = GetInput(operation, operands, kAuxInputTensor); in BidirectionalSequenceLSTM()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | BidirectionalSequenceRNN.h | 37 constexpr uint32_t kAuxInputTensor = 9; // optional variable
|
| D | BidirectionalSequenceLSTM.h | 117 static constexpr int kAuxInputTensor = 39; // Optional variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | bidirectional_sequence_rnn.cc | 57 constexpr int kAuxInputTensor = 9; // Optional. variable
130 GetOptionalInputTensor(context, node, kAuxInputTensor); in Prepare()
741 GetOptionalInputTensor(context, node, kAuxInputTensor); in Eval()
|
| H A D | bidirectional_sequence_lstm.cc | 113 constexpr int kAuxInputTensor = 39; // Optional variable
490 GetOptionalInputTensor(context, node, kAuxInputTensor); in Prepare()
1126 GetOptionalInputTensor(context, node, kAuxInputTensor); in Eval()
|