Searched refs:CheckInputTensorDimensions (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| 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()
599 context, CheckInputTensorDimensions(context, node, n_input, n_bw_output, in Prepare()
|
| D | unidirectional_sequence_lstm.cc | 409 TfLiteStatus CheckInputTensorDimensions(TfLiteContext* context, in CheckInputTensorDimensions() function
909 context, CheckInputTensorDimensions(context, node, n_input, n_output, in Prepare()
|
| D | lstm.cc | 852 TfLiteStatus CheckInputTensorDimensions(TfLiteContext* context, in CheckInputTensorDimensions() function
1343 context, CheckInputTensorDimensions(context, node, n_input, n_output, in Prepare()
|