Searched defs: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()
500 Value* cell_gate; in ParseLSTMAttributes() local
|
| /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
|
| /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()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 220 const float* cell_gate, bool use_cifg, float clip) { in UpdateLstmCellFloat()
573 int16_t* forget_gate, const int16_t* cell_gate, in UpdateLstmCellInteger()
|