Searched defs:input_gate (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | lstm_parser.cc | 275 Value* forget_gate, Value* input_gate, in BuildCellStateUpdate()
493 Value* input_gate; in ParseLSTMAttributes() local
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NELSTMLayer.cpp | 462 …TensorInfo input_gate = TensorInfo(TensorShape(num_cells, num_batches), 1, input->data_type()… in validate() local
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | LSTMLayerFixture.h | 426 SimpleTensor<T> input_gate; in compute_reference() local
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLLSTMLayer.cpp | 493 …TensorInfo input_gate = TensorInfo(TensorShape(num_cells, num_batches), 1, input->data_type()… in validate() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 101 const float* input_gate, float* forget_gate, in UpdateLstmCellFloat()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 219 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/ |
| D | reference_ops.h | 531 const float input_gate = in LstmCell() local
|