Searched refs:scratch5 (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | lstm_eval.cc | 513 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 …]
|
D | lstm_eval.h | 195 TfLiteTensor* scratch3, TfLiteTensor* scratch4, TfLiteTensor* scratch5, 221 TfLiteTensor* scratch3, TfLiteTensor* scratch4, TfLiteTensor* scratch5,
|
D | lstm.cc | 2092 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()
|
D | unidirectional_sequence_lstm.cc | 1413 TfLiteTensor* scratch5; in Eval() local 1415 GetTemporarySafe(context, node, 5, &scratch5)); in Eval() 1428 output, scratch0, scratch1, scratch2, scratch3, scratch4, scratch5, in Eval()
|