Searched refs:cell_state_ptr (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 65 float* output_state_ptr, float* cell_state_ptr, float* input_gate_scratch, in LstmStepWithAuxInput()
200 cell_to_input_weights_ptr, n_cell, cell_state_ptr, n_batch, in LstmStepWithAuxInput()
221 cell_to_forget_weights_ptr, n_cell, cell_state_ptr, n_batch, in LstmStepWithAuxInput()
239 tensor_utils::VectorVectorCwiseProduct(forget_gate_scratch, cell_state_ptr, in LstmStepWithAuxInput()
240 n_batch * n_cell, cell_state_ptr); in LstmStepWithAuxInput()
258 cell_scratch, forget_gate_scratch, n_batch * n_cell, cell_state_ptr); in LstmStepWithAuxInput()
261 cell_scratch, input_gate_scratch, n_batch * n_cell, cell_state_ptr); in LstmStepWithAuxInput()
264 tensor_utils::CwiseClipping(cell_state_ptr, n_batch * n_cell, in LstmStepWithAuxInput()
271 cell_to_output_weights_ptr, n_cell, cell_state_ptr, n_batch, in LstmStepWithAuxInput()
287 tensor_utils::ApplyActivationToVector(cell_state_ptr, n_batch * n_cell, in LstmStepWithAuxInput()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 833 float* output_state_ptr, float* cell_state_ptr, float* scratch0, in LstmStepFloat()
857 recurrent_to_input_weights_ptr, cell_state_ptr, in LstmStepFloat()
867 recurrent_to_forget_weights_ptr, cell_state_ptr, in LstmStepFloat()
882 UpdateLstmCellFloat(n_batch, n_cell, cell_state_ptr, input_gate_scratch, in LstmStepFloat()
889 recurrent_to_output_weights_ptr, cell_state_ptr, in LstmStepFloat()
895 CalculateLstmOutputFloat(n_batch, n_cell, n_output, cell_state_ptr, in LstmStepFloat()
1031 float* output_state_ptr, float* cell_state_ptr, int32_t* accum_scratch_ptr, in LstmStepHybrid()
1148 cell_state_ptr, cell_to_input_weights_ptr, cell_to_input_weights_scale, in LstmStepHybrid()
1165 cell_state_ptr, cell_to_forget_weights_ptr, cell_to_forget_weights_scale, in LstmStepHybrid()
1189 UpdateLstmCellFloat(n_batch, n_cell, cell_state_ptr, input_gate_scratch, in LstmStepHybrid()
[all …]
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 182 float* output_state_ptr, float* cell_state_ptr, float* scratch0, in LstmStepCalibration()
208 recurrent_to_input_weights_ptr, cell_state_ptr, in LstmStepCalibration()
219 recurrent_to_forget_weights_ptr, cell_state_ptr, in LstmStepCalibration()
236 UpdateLstmCellFloat(n_batch, n_cell, cell_state_ptr, input_gate_scratch, in LstmStepCalibration()
243 recurrent_to_output_weights_ptr, cell_state_ptr, in LstmStepCalibration()
251 n_batch, n_cell, n_output, cell_state_ptr, output_gate_scratch, in LstmStepCalibration()
402 float* cell_state_ptr = GetTensorData<float>(cell_state) + b * n_cell; in EvalCalibration() local
437 output_state_ptr, cell_state_ptr, input_gate_scratch_ptr, in EvalCalibration()
|