Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.h195 TfLiteTensor* scratch3, TfLiteTensor* scratch4, TfLiteTensor* scratch5,
221 TfLiteTensor* scratch3, TfLiteTensor* scratch4, TfLiteTensor* scratch5,
Dlstm_eval.cc440 float* scratch2, int32_t* scratch3, int32_t* scratch4) { in CalculateLstmOutputHybrid() argument
473 /*per_channel_scale=*/nullptr, scratch3, scratch4, in CalculateLstmOutputHybrid()
1379 int8_t* scratch4, int32_t* scratch5, CpuBackendContext* context) { in LstmStepInteger8x8_16()
1468 quantized_proj_clip, output_state_ptr, context, scratch0, scratch4, in LstmStepInteger8x8_16()
1622 int16_t* scratch4, int16_t* scratch5, int16_t* scratch6, in LstmStepInteger8x8_8()
1630 int16_t* output_gate_scratch = scratch4; in LstmStepInteger8x8_8()
2173 TfLiteTensor* scratch3, TfLiteTensor* scratch4, TfLiteTensor* scratch5, in EvalInteger8x8_16() argument
2281 GetTensorData<int8_t>(scratch4), GetTensorData<int32_t>(scratch5), in EvalInteger8x8_16()
2379 GetTensorData<int16_t>(scratch3), GetTensorData<int8_t>(scratch4), in EvalInteger8x8_16()
2411 TfLiteTensor* scratch3, TfLiteTensor* scratch4, TfLiteTensor* scratch5, in EvalInteger8x8_8() argument
[all …]
Dlstm.cc2089 TfLiteTensor* scratch4; in Eval() local
2091 GetTemporarySafe(context, node, 4, &scratch4)); in Eval()
2108 scratch4, scratch5, CpuBackendContext::GetFromContext(context)); in Eval()
2122 TfLiteTensor* scratch4; in Eval() local
2124 GetTemporarySafe(context, node, 4, &scratch4)); in Eval()
2146 scratch3, scratch4, scratch5, scratch6, scratch7); in Eval()
Dunidirectional_sequence_lstm.cc1410 TfLiteTensor* scratch4; in Eval() local
1412 GetTemporarySafe(context, node, 4, &scratch4)); in Eval()
1428 output, scratch0, scratch1, scratch2, scratch3, scratch4, scratch5, in Eval()