Searched refs:cell_gate (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | lstm_parser.cc | 252 Value* cell_gate, float cell_clip, in BuildCellStateUpdate() argument
257 Value* cell_state_contrib = CreateNewSimilarValue(graph, cell_gate); in BuildCellStateUpdate()
266 Value* cell_gate_contrib = CreateNewSimilarValue(graph, cell_gate); in BuildCellStateUpdate()
273 RETURN_IF_ERROR(graph->AddConsumer(node->id, cell_gate->id)); in BuildCellStateUpdate()
276 Value* new_cell_state = CreateNewSimilarValue(graph, cell_gate); in BuildCellStateUpdate()
500 Value* cell_gate; in ParseLSTMAttributes() local
509 &cell_gate)); in ParseLSTMAttributes()
513 cell_gate, params->cell_clip, in ParseLSTMAttributes()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | load_quantization_recipe.cc | 165 auto cell_gate = CreateGate(loc, lstm.input(), lstm.input_to_cell_weights(), in LoadForLSTMOp() local
174 loc, int16, input_gate->getResult(0), cell_gate->getResult(0), none_af); in LoadForLSTMOp()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 102 const float* cell_gate, bool use_cifg, float clip) { in UpdateLstmCellFloat() argument
113 cell_gate, scratch, n_batch * n_cell, cell_state); in UpdateLstmCellFloat()
116 cell_gate, input_gate, n_batch * n_cell, cell_state); in UpdateLstmCellFloat()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 220 const float* cell_gate, bool use_cifg, float clip) { in UpdateLstmCellFloat() argument
231 cell_gate, scratch, n_batch * n_cell, cell_state); in UpdateLstmCellFloat()
234 cell_gate, input_gate, n_batch * n_cell, cell_state); in UpdateLstmCellFloat()
573 int16_t* forget_gate, const int16_t* cell_gate, in UpdateLstmCellInteger() argument
584 tensor_utils::CwiseMul(scratch, cell_gate, n_batch, n_cell, in UpdateLstmCellInteger()
587 tensor_utils::CwiseMul(input_gate, cell_gate, n_batch, n_cell, in UpdateLstmCellInteger()
|