Home
last modified time | relevance | path

Searched refs:num_intermediate_tensors (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dlstm.cc1377 const int num_intermediate_tensors = node->intermediates->size; in Prepare() local
1379 TF_LITE_ENSURE(context, num_intermediate_tensors == 5 || in Prepare()
1380 num_intermediate_tensors == 12); in Prepare()
1384 const bool is_8x8_16 = num_intermediate_tensors == 5; in Prepare()
2079 const int num_intermediate_tensors = node->intermediates->size; in Eval() local
2092 if (num_intermediate_tensors == 5) { in Eval()
H A Dunidirectional_sequence_lstm.cc947 const int num_intermediate_tensors = node->intermediates->size; in Prepare() local
948 TF_LITE_ENSURE(context, num_intermediate_tensors == 5); in Prepare()