Searched refs:cell_scratch (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 119 float* forget_gate_scratch, float* cell_scratch, float* output_gate_scratch, in LstmStepWithAuxInput()
135 tensor_utils::ZeroVector(cell_scratch, n_cell * n_batch); in LstmStepWithAuxInput()
145 cell_scratch); in LstmStepWithAuxInput()
162 cell_scratch, /*result_stride=*/1); in LstmStepWithAuxInput()
181 n_batch, cell_scratch, /*result_stride=*/1); in LstmStepWithAuxInput()
199 n_batch, cell_scratch, /*result_stride=*/1); in LstmStepWithAuxInput()
249 tensor_utils::MeanStddevNormalization(cell_scratch, cell_scratch, n_cell, in LstmStepWithAuxInput()
252 cell_layer_norm_coefficients_ptr, n_cell, cell_scratch, n_batch, in LstmStepWithAuxInput()
253 cell_scratch); in LstmStepWithAuxInput()
255 cell_scratch); in LstmStepWithAuxInput()
[all …]
|