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.cc74 kNumTemporaryTensors = 5 enumerator
79 context->AddTensors(context, kNumTemporaryTensors, scratch_tensor_index); in Init()
180 node->temporaries = TfLiteIntArrayCreate(kNumTemporaryTensors); in Prepare()
183 node->temporaries = TfLiteIntArrayCreate(kNumTemporaryTensors - 1); in Prepare()
Dunidirectional_sequence_lstm.cc86 kNumTemporaryTensors = 7 enumerator
91 context->AddTensors(context, kNumTemporaryTensors, scratch_tensor_index); in Init()
315 node->temporaries = TfLiteIntArrayCreate(kNumTemporaryTensors); in Prepare()
Dbidirectional_sequence_lstm.cc143 kNumTemporaryTensors = 11 enumerator
148 context->AddTensors(context, kNumTemporaryTensors, scratch_tensor_index); in Init()
518 has_aux_input ? kNumTemporaryTensors : kNumTemporaryTensors - 1); in Prepare()