Searched defs:output_gate (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | lstm_parser.cc | 337 Value* output_state, Value* output_gate, in BuildOutputStateUpdate()
516 Value* output_gate; in ParseLSTMAttributes() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | load_quantization_recipe.cc | 178 auto output_gate = CreateGate( in LoadForLSTMOp() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 263 const float* cell_state, const float* output_gate, in CalculateLstmOutputFloat()
434 const float* output_gate, TfLiteFusedActivation activation, in CalculateLstmOutputHybrid()
618 int32_t cell_state_scale, const int16_t* output_gate, in CalculateLstmOutputInteger8x8_16()
726 const int16_t* output_gate, const int8_t* projection_weights, in CalculateLstmOutputInteger8x8_8()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 125 const float* output_gate, TfLiteFusedActivation activation, in CalculateLstmOutputCalibration()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 539 const float output_gate = in LstmCell() local
|