Home
last modified time | relevance | path

Searched refs:scratch5 (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.cc513 int32_t* scratch5) { in CalculateLstmGateInteger8x8_16()
523 input_to_gate_scale_b, n_batch, n_input, n_cell, 0, scratch5, gate, in CalculateLstmGateInteger8x8_16()
531 n_cell, 0, scratch5, gate, context); in CalculateLstmGateInteger8x8_16()
1379 int8_t* scratch4, int32_t* scratch5, CpuBackendContext* context) { in LstmStepInteger8x8_16()
1418 kTfLiteActSigmoid, input_gate_scratch, context, scratch5); in LstmStepInteger8x8_16()
1432 scratch5); in LstmStepInteger8x8_16()
1444 cell_gate_scratch, context, scratch5); in LstmStepInteger8x8_16()
1461 scratch5); in LstmStepInteger8x8_16()
1469 scratch5); in LstmStepInteger8x8_16()
1622 int16_t* scratch4, int16_t* scratch5, int16_t* scratch6, in LstmStepInteger8x8_8()
[all …]
Dlstm_eval.h195 TfLiteTensor* scratch3, TfLiteTensor* scratch4, TfLiteTensor* scratch5,
221 TfLiteTensor* scratch3, TfLiteTensor* scratch4, TfLiteTensor* scratch5,
Dlstm.cc2092 TfLiteTensor* scratch5; in Eval() local
2094 GetTemporarySafe(context, node, 5, &scratch5)); in Eval()
2108 scratch4, scratch5, CpuBackendContext::GetFromContext(context)); in Eval()
2125 TfLiteTensor* scratch5; in Eval() local
2127 GetTemporarySafe(context, node, 5, &scratch5)); in Eval()
2146 scratch3, scratch4, scratch5, scratch6, scratch7); in Eval()
Dunidirectional_sequence_lstm.cc1413 TfLiteTensor* scratch5; in Eval() local
1415 GetTemporarySafe(context, node, 5, &scratch5)); in Eval()
1428 output, scratch0, scratch1, scratch2, scratch3, scratch4, scratch5, in Eval()