Searched refs:output_state_sf (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 310 const int8_t* output_state, const float* output_state_sf, in CalculateLstmGateHybrid() argument
388 recurrent_to_gate_weights_scale, output_state_sf, n_batch, gate, in CalculateLstmGateHybrid()
1027 float* input_sf, float* aux_input_sf, float* output_state_sf, in LstmStepHybrid()
1135 output_state_sf, output_state_zp, asymmetric_quantize_inputs); in LstmStepHybrid()
1145 quantized_output_state_ptr, output_state_sf, output_state_zp, in LstmStepHybrid()
1162 quantized_output_state_ptr, output_state_sf, output_state_zp, in LstmStepHybrid()
1178 quantized_output_state_ptr, output_state_sf, output_state_zp, in LstmStepHybrid()
1199 quantized_output_state_ptr, output_state_sf, output_state_zp, in LstmStepHybrid()
1904 TfLiteTensor* aux_input_sf, TfLiteTensor* output_state_sf, in EvalHybrid()
2033 GetTensorData<float>(output_state_sf), in EvalHybrid()
[all …]
|
| D | bidirectional_sequence_lstm.cc | 777 TfLiteTensor* output_state_sf; in Prepare() local
780 &output_state_sf)); in Prepare()
781 output_state_sf->type = kTfLiteFloat32; in Prepare()
782 output_state_sf->allocation_type = kTfLiteArenaRw; in Prepare()
783 if (!TfLiteIntArrayEqualsArray(output_state_sf->dims, 1, scaling_dims)) { in Prepare()
786 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, output_state_sf, in Prepare()
|
| D | unidirectional_sequence_lstm.cc | 1044 TfLiteTensor* output_state_sf; in Prepare() local
1047 &output_state_sf)); in Prepare()
1048 output_state_sf->type = kTfLiteFloat32; in Prepare()
1049 output_state_sf->allocation_type = kTfLiteArenaRw; in Prepare()
1050 if (!TfLiteIntArrayEqualsArray(output_state_sf->dims, 1, scaling_dims)) { in Prepare()
1053 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, output_state_sf, in Prepare()
|
| D | lstm_eval.h | 162 TfLiteTensor* aux_input_sf, TfLiteTensor* output_state_sf,
|
| D | lstm.cc | 1502 TfLiteTensor* output_state_sf; in Prepare() local
1505 &output_state_sf)); in Prepare()
1506 output_state_sf->type = kTfLiteFloat32; in Prepare()
1507 output_state_sf->allocation_type = kTfLiteArenaRw; in Prepare()
1508 if (!TfLiteIntArrayEqualsArray(output_state_sf->dims, 1, scaling_dims)) { in Prepare()
1511 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, output_state_sf, in Prepare()
|