Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.cc833 float* output_state_ptr, float* cell_state_ptr, float* scratch0, in LstmStepFloat()
856 aux_input_to_input_weights_ptr, output_state_ptr, in LstmStepFloat()
866 aux_input_to_forget_weights_ptr, output_state_ptr, in LstmStepFloat()
874 aux_input_to_cell_weights_ptr, output_state_ptr, in LstmStepFloat()
888 aux_input_to_output_weights_ptr, output_state_ptr, in LstmStepFloat()
898 params->proj_clip, output_state_ptr, scratch2); in LstmStepFloat()
902 std::copy_n(output_state_ptr + b * n_output, n_output, in LstmStepFloat()
1031 float* output_state_ptr, float* cell_state_ptr, int32_t* accum_scratch_ptr, in LstmStepHybrid()
1120 tensor_utils::IsZeroVector(output_state_ptr, n_batch * n_output); in LstmStepHybrid()
1134 output_state_ptr, n_batch, n_output, quantized_output_state_ptr, in LstmStepHybrid()
[all …]
Dlstm.cc568 int8_t* output_state_ptr = nullptr; in PopulateQuantizedLstmParams8x8_8() local
666 output_state_ptr = output_state->data.int8; in PopulateQuantizedLstmParams8x8_8()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc65 float* output_state_ptr, float* cell_state_ptr, float* input_gate_scratch, in LstmStepWithAuxInput()
151 recurrent_to_input_weights_ptr, n_cell, n_output, output_state_ptr, in LstmStepWithAuxInput()
155 recurrent_to_forget_weights_ptr, n_cell, n_output, output_state_ptr, in LstmStepWithAuxInput()
158 recurrent_to_cell_weights_ptr, n_cell, n_output, output_state_ptr, in LstmStepWithAuxInput()
161 recurrent_to_output_weights_ptr, n_cell, n_output, output_state_ptr, in LstmStepWithAuxInput()
168 recurrent_to_input_weights_ptr, n_cell, n_output, output_state_ptr, in LstmStepWithAuxInput()
175 recurrent_to_forget_weights_ptr, n_cell, n_output, output_state_ptr, in LstmStepWithAuxInput()
182 recurrent_to_cell_weights_ptr, n_cell, n_output, output_state_ptr, in LstmStepWithAuxInput()
190 recurrent_to_output_weights_ptr, n_cell, n_output, output_state_ptr, in LstmStepWithAuxInput()
327 output_state_ptr + k * n_output); in LstmStepWithAuxInput()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc182 float* output_state_ptr, float* cell_state_ptr, float* scratch0, in LstmStepCalibration()
207 aux_input_to_input_weights_ptr, output_state_ptr, in LstmStepCalibration()
218 aux_input_to_forget_weights_ptr, output_state_ptr, in LstmStepCalibration()
227 aux_input_to_cell_weights_ptr, output_state_ptr, in LstmStepCalibration()
242 aux_input_to_output_weights_ptr, output_state_ptr, in LstmStepCalibration()
253 params->proj_clip, output_state_ptr, scratch2, logger, in LstmStepCalibration()
258 std::copy_n(output_state_ptr + b * n_output, n_output, in LstmStepCalibration()
400 float* output_state_ptr = in EvalCalibration() local
437 output_state_ptr, cell_state_ptr, input_gate_scratch_ptr, in EvalCalibration()