Home
last modified time | relevance | path

Searched defs:input_gate (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dlstm_parser.cc251 Value* forget_gate, Value* input_gate, in BuildCellStateUpdate()
475 Value* input_gate; in ParseLSTMAttributes() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dload_quantization_recipe.cc151 auto input_gate = CreateGate( in LoadForLSTMOp() local
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc101 const float* input_gate, float* forget_gate, in UpdateLstmCellFloat()
/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.cc219 const float* input_gate, float* forget_gate, in UpdateLstmCellFloat()
572 int32_t cell_state_scale, const int16_t* input_gate, in UpdateLstmCellInteger()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h529 const float input_gate = in LstmCell() local