Searched refs:CheckInputTensorDimensions (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_lstm.cc | 100 TfLiteStatus CheckInputTensorDimensions(TfLiteContext* context, in CheckInputTensorDimensions() function
284 TF_LITE_ENSURE_OK(context, CheckInputTensorDimensions(context, node, n_input, in Prepare()
|
| D | bidirectional_sequence_lstm.cc | 349 TfLiteStatus CheckInputTensorDimensions(TfLiteContext* context, in CheckInputTensorDimensions() function
442 context, CheckInputTensorDimensions(context, node, n_input, n_fw_output, in Prepare()
551 context, CheckInputTensorDimensions(context, node, n_input, n_bw_output, in Prepare()
|
| D | lstm.cc | 116 TfLiteStatus CheckInputTensorDimensions(TfLiteContext* context, in CheckInputTensorDimensions() function
359 CheckInputTensorDimensions(context, node, n_input, n_output, in Prepare()
|