Home
last modified time | relevance | path

Searched refs:kNumTemporaryTensors (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dbidirectional_sequence_rnn.cc77 kNumTemporaryTensors = 9 enumerator
82 context->AddTensors(context, kNumTemporaryTensors, in Init()
194 node->temporaries = TfLiteIntArrayCreate(kNumTemporaryTensors); in Prepare()
197 node->temporaries = TfLiteIntArrayCreate(kNumTemporaryTensors - 1); in Prepare()
Dbidirectional_sequence_lstm.cc154 kNumTemporaryTensors = 19, enumerator
165 context->AddTensors(context, kNumTemporaryTensors, in Init()
564 has_aux_input ? kNumTemporaryTensors : kNumTemporaryTensors - 1); in Prepare()
Dunidirectional_sequence_lstm.cc394 kNumTemporaryTensors = 12, enumerator
399 context->AddTensors(context, kNumTemporaryTensors, in Init()
943 node->temporaries = TfLiteIntArrayCreate(kNumTemporaryTensors); in Prepare()