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()
|