Searched refs:CheckInputTensorDimensions (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | LSTM.h | 183 static bool CheckInputTensorDimensions(
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BidirectionalSequenceLSTM.cpp | 287 if (!LSTMCell::CheckInputTensorDimensions( in Prepare()
395 if (!LSTMCell::CheckInputTensorDimensions( in Prepare()
|
| D | LSTM.cpp | 132 bool LSTMCell::CheckInputTensorDimensions( in CheckInputTensorDimensions() function in android::nn::LSTMCell
360 if (!CheckInputTensorDimensions( in Prepare()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | bidirectional_sequence_lstm.cc | 383 TfLiteStatus CheckInputTensorDimensions(TfLiteContext* context, in CheckInputTensorDimensions() function
485 context, CheckInputTensorDimensions(context, node, n_input, n_fw_output, in Prepare()
601 context, CheckInputTensorDimensions(context, node, n_input, n_bw_output, in Prepare()
|
| H A D | unidirectional_sequence_lstm.cc | 417 TfLiteStatus CheckInputTensorDimensions(TfLiteContext* context, in CheckInputTensorDimensions() function
919 context, CheckInputTensorDimensions(context, node, n_input, n_output, in Prepare()
|
| H A D | lstm.cc | 853 TfLiteStatus CheckInputTensorDimensions(TfLiteContext* context, in CheckInputTensorDimensions() function
1344 context, CheckInputTensorDimensions(context, node, n_input, n_output, in Prepare()
|