Searched defs:input_gate (Results 1 – 8 of 8) 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()
493 Value* input_gate; in ParseLSTMAttributes() local
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
| H A D | NELSTMLayer.cpp | 460 …TensorInfo input_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 | 485 …TensorInfo input_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 | 432 SimpleTensor<T> input_gate; in compute_reference() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | lstm_cell.h | 116 const float input_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()
614 int32_t cell_state_scale, const int16_t* input_gate, in UpdateLstmCellInteger()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | RNN.cpp | 784 const auto input_gate = in operator ()() local
|