Searched defs:forget_gate (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | lstm_parser.cc | 275 Value* forget_gate, Value* input_gate, in BuildCellStateUpdate()
483 Value* forget_gate; in ParseLSTMAttributes() local
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
| H A D | NELSTMLayer.cpp | 461 …TensorInfo forget_gate = TensorInfo(TensorShape(num_cells, num_batches), 1, input->data_type()… in validate() local
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
| H A D | CLLSTMLayer.cpp | 486 …TensorInfo forget_gate = TensorInfo(TensorShape(num_cells, num_batches), 1, input->data_type()… in validate() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | LSTMLayerFixture.h | 412 …SimpleTensor<T> forget_gate = reference::arithmetic_operation(reference::ArithmeticOper… in compute_reference() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | lstm_cell.h | 122 const float forget_gate = in LstmCell() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 101 const float* input_gate, float* forget_gate, in UpdateLstmCellFloat()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_eval.cc | 256 const float* input_gate, float* forget_gate, in UpdateLstmCellFloat()
615 int16_t* forget_gate, const int16_t* cell_gate, in UpdateLstmCellInteger()
|