Searched defs:output_gate (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | lstm_parser.cc | 361 Value* output_state, Value* output_gate, in BuildOutputStateUpdate()
534 Value* output_gate; in ParseLSTMAttributes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_eval.cc | 303 const float* cell_state, const float* output_gate, in CalculateLstmOutputFloat()
476 const float* output_gate, TfLiteFusedActivation activation, in CalculateLstmOutputHybrid()
660 int32_t cell_state_scale, const int16_t* output_gate, in CalculateLstmOutputInteger8x8_16()
768 const int16_t* output_gate, const int8_t* projection_weights, in CalculateLstmOutputInteger8x8_8()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | lstm_cell.h | 126 const float output_gate = in LstmCell() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 125 const float* output_gate, TfLiteFusedActivation activation, in CalculateLstmOutputCalibration()
|