Home
last modified time | relevance | path

Searched refs:CheckInputTensorDimensions (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/
DLSTM.h183 static bool CheckInputTensorDimensions(
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DBidirectionalSequenceLSTM.cpp287 if (!LSTMCell::CheckInputTensorDimensions( in Prepare()
395 if (!LSTMCell::CheckInputTensorDimensions( in Prepare()
DLSTM.cpp132 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 Dbidirectional_sequence_lstm.cc383 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 Dunidirectional_sequence_lstm.cc417 TfLiteStatus CheckInputTensorDimensions(TfLiteContext* context, in CheckInputTensorDimensions() function
919 context, CheckInputTensorDimensions(context, node, n_input, n_output, in Prepare()
H A Dlstm.cc853 TfLiteStatus CheckInputTensorDimensions(TfLiteContext* context, in CheckInputTensorDimensions() function
1344 context, CheckInputTensorDimensions(context, node, n_input, n_output, in Prepare()